mirror of
https://github.com/apache/arrow.git
synced 2026-10-29 00:56:31 +08:00
tianchenandAntoine Pitrou
843e8bb556
Closes #7887 from tianchen92/ARROW-9304 Lead-authored-by: tianchen <tianchen@apache.org> Co-authored-by: Antoine Pitrou <antoine@python.org> Signed-off-by: Antoine Pitrou <antoine@python.org>
95 lines
2.9 KiB
Ruby
95 lines
2.9 KiB
Ruby
# 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.
|
|
|
|
class StructArrayTest < Test::Unit::TestCase
|
|
sub_test_case(".new") do
|
|
test("build") do
|
|
data_type = Arrow::StructDataType.new(visible: :boolean,
|
|
count: :uint64)
|
|
values = [
|
|
[true, 1],
|
|
nil,
|
|
[false, 2],
|
|
]
|
|
array = Arrow::StructArray.new(data_type, values)
|
|
assert_equal([
|
|
[true, false, false],
|
|
[1, 0, 2],
|
|
],
|
|
[
|
|
array.find_field(0).to_a,
|
|
array.find_field(1).to_a,
|
|
])
|
|
end
|
|
end
|
|
|
|
sub_test_case("instance methods") do
|
|
def setup
|
|
@data_type = Arrow::StructDataType.new(visible: {type: :boolean},
|
|
count: {type: :uint64})
|
|
@values = [
|
|
[true, 1],
|
|
[false, 2],
|
|
]
|
|
@array = Arrow::StructArray.new(@data_type, @values)
|
|
end
|
|
|
|
test("#[]") do
|
|
assert_equal([
|
|
{"visible" => true, "count" => 1},
|
|
{"visible" => false, "count" => 2},
|
|
],
|
|
@array.to_a)
|
|
end
|
|
|
|
test("#get_value") do
|
|
assert_equal([
|
|
{"visible" => true, "count" => 1},
|
|
{"visible" => false, "count" => 2},
|
|
],
|
|
[
|
|
@array.get_value(0),
|
|
@array.get_value(1),
|
|
])
|
|
end
|
|
|
|
sub_test_case("#find_field") do
|
|
test("Integer") do
|
|
assert_equal([
|
|
[true, false],
|
|
[1, 2],
|
|
],
|
|
[
|
|
@array.find_field(0).to_a,
|
|
@array.find_field(1).to_a,
|
|
])
|
|
end
|
|
|
|
test("String, Symbol") do
|
|
assert_equal([
|
|
[true, false],
|
|
[1, 2],
|
|
],
|
|
[
|
|
@array.find_field("visible").to_a,
|
|
@array.find_field(:count).to_a,
|
|
])
|
|
end
|
|
end
|
|
end
|
|
end
|