test/test-chunked-array.rb in red-arrow-0.16.0 vs test/test-chunked-array.rb in red-arrow-0.17.0
- old
+ new
@@ -90,32 +90,34 @@
arrays = [
Arrow::BooleanArray.new([false, true]),
Arrow::BooleanArray.new([false, true, false]),
]
@chunked_array = Arrow::ChunkedArray.new(arrays)
+ @options = Arrow::FilterOptions.new
+ @options.null_selection_behavior = :emit_null
end
test("Array: boolean") do
filter = [nil, true, true, false, true]
chunks = [
Arrow::BooleanArray.new([nil, true]),
Arrow::BooleanArray.new([false, false]),
]
filtered_chunked_array = Arrow::ChunkedArray.new(chunks)
assert_equal(filtered_chunked_array,
- @chunked_array.filter(filter))
+ @chunked_array.filter(filter, @options))
end
test("Arrow::BooleanArray") do
filter = Arrow::BooleanArray.new([nil, true, true, false, true])
chunks = [
Arrow::BooleanArray.new([nil, true]),
Arrow::BooleanArray.new([false, false]),
]
filtered_chunked_array = Arrow::ChunkedArray.new(chunks)
assert_equal(filtered_chunked_array,
- @chunked_array.filter(filter))
+ @chunked_array.filter(filter, @options))
end
test("Arrow::ChunkedArray") do
chunks = [
Arrow::BooleanArray.new([nil, true]),
@@ -126,10 +128,10 @@
Arrow::BooleanArray.new([nil, true]),
Arrow::BooleanArray.new([false, false]),
]
filtered_chunked_array = Arrow::ChunkedArray.new(filtered_chunks)
assert_equal(filtered_chunked_array,
- @chunked_array.filter(filter))
+ @chunked_array.filter(filter, @options))
end
end
sub_test_case("#take") do
def setup