| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, |
| // software distributed under the License is distributed on an |
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| // KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations |
| // under the License. |
| |
| import XCTest |
| @testable import Arrow |
| |
| final class ArrayTests: XCTestCase { |
| func testPrimitiveArray() throws { |
| // This is an example of a functional test case. |
| // Use XCTAssert and related functions to verify your tests produce the correct |
| // results. |
| let arrayBuilder: NumberArrayBuilder<UInt8> = try ArrowArrayBuilders.loadNumberArrayBuilder(); |
| for i in 0..<100 { |
| arrayBuilder.append(UInt8(i)) |
| } |
| |
| XCTAssertEqual(arrayBuilder.nullCount, 0) |
| arrayBuilder.append(nil) |
| XCTAssertEqual(arrayBuilder.length, 101) |
| XCTAssertEqual(arrayBuilder.capacity, 136) |
| XCTAssertEqual(arrayBuilder.nullCount, 1) |
| let array = try arrayBuilder.finish() |
| XCTAssertEqual(array.length, 101) |
| XCTAssertEqual(array[1]!, 1) |
| XCTAssertEqual(array[10]!, 10) |
| XCTAssertEqual(try array.isNull(100), true) |
| |
| let doubleBuilder: NumberArrayBuilder<Double> = try ArrowArrayBuilders.loadNumberArrayBuilder(); |
| doubleBuilder.append(14) |
| doubleBuilder.append(40.4) |
| XCTAssertEqual(doubleBuilder.nullCount, 0) |
| XCTAssertEqual(doubleBuilder.length, 2) |
| XCTAssertEqual(doubleBuilder.capacity, 264) |
| let doubleArray = try doubleBuilder.finish() |
| XCTAssertEqual(doubleArray.length, 2) |
| XCTAssertEqual(doubleArray[0]!, 14) |
| XCTAssertEqual(doubleArray[1]!, 40.4) |
| } |
| |
| func testStringArray() throws { |
| let stringBuilder = try ArrowArrayBuilders.loadStringArrayBuilder(); |
| for i in 0..<100 { |
| if i % 10 == 9 { |
| stringBuilder.append(nil) |
| } else { |
| stringBuilder.append("test" + String(i)) |
| } |
| } |
| XCTAssertEqual(stringBuilder.nullCount, 10) |
| XCTAssertEqual(stringBuilder.length, 100) |
| XCTAssertEqual(stringBuilder.capacity, 648) |
| let stringArray = try stringBuilder.finish() |
| XCTAssertEqual(stringArray.length, 100) |
| for i in 0..<stringArray.length { |
| if i % 10 == 9 { |
| XCTAssertEqual(try stringArray.isNull(i), true) |
| } else { |
| XCTAssertEqual(stringArray[i]!, "test" + String(i)) |
| } |
| } |
| |
| XCTAssertEqual(stringArray[1]!, "test1") |
| XCTAssertEqual(stringArray[0]!, "test0") |
| } |
| |
| func testBoolArray() throws { |
| let boolBuilder = try ArrowArrayBuilders.loadBoolArrayBuilder() |
| boolBuilder.append(true) |
| boolBuilder.append(nil) |
| boolBuilder.append(false) |
| boolBuilder.append(false) |
| XCTAssertEqual(boolBuilder.nullCount, 1) |
| XCTAssertEqual(boolBuilder.length, 4) |
| XCTAssertEqual(boolBuilder.capacity, 72) |
| let boolArray = try boolBuilder.finish() |
| XCTAssertEqual(boolArray.length, 4) |
| XCTAssertEqual(boolArray[1], nil) |
| XCTAssertEqual(boolArray[0]!, true) |
| XCTAssertEqual(boolArray[2]!, false) |
| |
| } |
| |
| func testDate32Array() throws { |
| let date32Builder: Date32ArrayBuilder = try ArrowArrayBuilders.loadDate32ArrayBuilder(); |
| let date2 = Date(timeIntervalSinceReferenceDate: 86400 * 1) |
| let date1 = Date(timeIntervalSinceReferenceDate: 86400 * 5000 + 352) |
| date32Builder.append(date1) |
| date32Builder.append(date2) |
| date32Builder.append(nil) |
| XCTAssertEqual(date32Builder.nullCount, 1) |
| XCTAssertEqual(date32Builder.length, 3) |
| XCTAssertEqual(date32Builder.capacity, 136) |
| let date32Array = try date32Builder.finish() |
| XCTAssertEqual(date32Array.length, 3) |
| XCTAssertEqual(date32Array[1], date2) |
| let adjustedDate1 = Date(timeIntervalSince1970: date1.timeIntervalSince1970 - 352) |
| XCTAssertEqual(date32Array[0]!, adjustedDate1) |
| } |
| |
| func testDate64Array() throws { |
| let date64Builder: Date64ArrayBuilder = try ArrowArrayBuilders.loadDate64ArrayBuilder(); |
| let date2 = Date(timeIntervalSinceReferenceDate: 86400 * 1) |
| let date1 = Date(timeIntervalSinceReferenceDate: 86400 * 5000 + 352) |
| date64Builder.append(date1) |
| date64Builder.append(date2) |
| date64Builder.append(nil) |
| XCTAssertEqual(date64Builder.nullCount, 1) |
| XCTAssertEqual(date64Builder.length, 3) |
| XCTAssertEqual(date64Builder.capacity, 264) |
| let date64Array = try date64Builder.finish() |
| XCTAssertEqual(date64Array.length, 3) |
| XCTAssertEqual(date64Array[1], date2) |
| XCTAssertEqual(date64Array[0]!, date1) |
| } |
| |
| func testBinaryArray() throws { |
| let binaryBuilder = try ArrowArrayBuilders.loadBinaryArrayBuilder(); |
| for i in 0..<100 { |
| if i % 10 == 9 { |
| binaryBuilder.append(nil) |
| } else { |
| binaryBuilder.append(("test" + String(i)).data(using:.utf8)) |
| } |
| } |
| |
| XCTAssertEqual(binaryBuilder.nullCount, 10) |
| XCTAssertEqual(binaryBuilder.length, 100) |
| XCTAssertEqual(binaryBuilder.capacity, 648) |
| let binaryArray = try binaryBuilder.finish() |
| XCTAssertEqual(binaryArray.length, 100) |
| for i in 0..<binaryArray.length { |
| if i % 10 == 9 { |
| XCTAssertEqual(try binaryArray.isNull(i), true) |
| } else { |
| let stringData = String(bytes: binaryArray[i]!, encoding: .utf8) |
| XCTAssertEqual(stringData, "test" + String(i)) |
| } |
| } |
| } |
| |
| func testTime32Array() throws { |
| let milliBuilder = try ArrowArrayBuilders.loadTime32ArrayBuilder(.Milliseconds); |
| milliBuilder.append(100) |
| milliBuilder.append(1000000) |
| milliBuilder.append(nil) |
| XCTAssertEqual(milliBuilder.nullCount, 1) |
| XCTAssertEqual(milliBuilder.length, 3) |
| XCTAssertEqual(milliBuilder.capacity, 136) |
| let milliArray = try milliBuilder.finish() |
| let milliType = milliArray.arrowData.type as! ArrowTypeTime32 |
| XCTAssertEqual(milliType.unit, .Milliseconds) |
| XCTAssertEqual(milliArray.length, 3) |
| XCTAssertEqual(milliArray[1], 1000000) |
| XCTAssertEqual(milliArray[2], nil) |
| |
| let secBuilder = try ArrowArrayBuilders.loadTime32ArrayBuilder(.Seconds); |
| secBuilder.append(200) |
| secBuilder.append(nil) |
| secBuilder.append(2000011) |
| XCTAssertEqual(secBuilder.nullCount, 1) |
| XCTAssertEqual(secBuilder.length, 3) |
| XCTAssertEqual(secBuilder.capacity, 136) |
| let secArray = try secBuilder.finish() |
| let secType = secArray.arrowData.type as! ArrowTypeTime32 |
| XCTAssertEqual(secType.unit, .Seconds) |
| XCTAssertEqual(secArray.length, 3) |
| XCTAssertEqual(secArray[1], nil) |
| XCTAssertEqual(secArray[2], 2000011) |
| } |
| |
| func testTime64Array() throws { |
| let nanoBuilder = try ArrowArrayBuilders.loadTime64ArrayBuilder(.Nanoseconds); |
| nanoBuilder.append(10000) |
| nanoBuilder.append(nil) |
| nanoBuilder.append(123456789) |
| XCTAssertEqual(nanoBuilder.nullCount, 1) |
| XCTAssertEqual(nanoBuilder.length, 3) |
| XCTAssertEqual(nanoBuilder.capacity, 264) |
| let nanoArray = try nanoBuilder.finish() |
| let nanoType = nanoArray.arrowData.type as! ArrowTypeTime64 |
| XCTAssertEqual(nanoType.unit, .Nanoseconds) |
| XCTAssertEqual(nanoArray.length, 3) |
| XCTAssertEqual(nanoArray[1], nil) |
| XCTAssertEqual(nanoArray[2], 123456789) |
| |
| let microBuilder = try ArrowArrayBuilders.loadTime64ArrayBuilder(.Microseconds); |
| microBuilder.append(nil) |
| microBuilder.append(20000) |
| microBuilder.append(987654321) |
| XCTAssertEqual(microBuilder.nullCount, 1) |
| XCTAssertEqual(microBuilder.length, 3) |
| XCTAssertEqual(microBuilder.capacity, 264) |
| let microArray = try microBuilder.finish() |
| let microType = microArray.arrowData.type as! ArrowTypeTime64 |
| XCTAssertEqual(microType.unit, .Microseconds) |
| XCTAssertEqual(microArray.length, 3) |
| XCTAssertEqual(microArray[1], 20000) |
| XCTAssertEqual(microArray[2], 987654321) |
| } |
| } |
