red-arrow 8.0.0 → 24.0.0

data/test/test-decimal128-array-builder.rb

@@ -1,112 +0,0 @@
-# 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 Decimal128ArrayBuilderTest < Test::Unit::TestCase
-  def setup
-    @data_type = Arrow::Decimal128DataType.new(3, 1)
-    @builder = Arrow::Decimal128ArrayBuilder.new(@data_type)
-  end
-
-  sub_test_case("#append_value") do
-    test("nil") do
-      @builder.append_value(nil)
-      array = @builder.finish
-      assert_equal(nil, array[0])
-    end
-
-    test("Arrow::Decimal128") do
-      @builder.append_value(Arrow::Decimal128.new("10.1"))
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-
-    test("String") do
-      @builder.append_value("10.1")
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-
-    test("Float") do
-      @builder.append_value(10.1)
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-
-    test("BigDecimal") do
-      @builder.append_value(BigDecimal("10.1"))
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-  end
-
-  sub_test_case("#append_values") do
-    test("mixed") do
-      @builder.append_values([
-                               Arrow::Decimal128.new("10.1"),
-                               nil,
-                               "10.1",
-                               10.1,
-                               BigDecimal("10.1"),
-                             ])
-      array = @builder.finish
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                     BigDecimal("10.1"),
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-
-    test("is_valids") do
-      @builder.append_values([
-                               Arrow::Decimal128.new("10.1"),
-                               Arrow::Decimal128.new("10.1"),
-                               Arrow::Decimal128.new("10.1"),
-                             ],
-                             [
-                               true,
-                               false,
-                               true,
-                             ])
-      array = @builder.finish
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-
-    test("packed") do
-      @builder.append_values(Arrow::Decimal128.new("10.1").to_bytes.to_s * 3,
-                             [true, false, true])
-      array = @builder.finish
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-  end
-end

data/test/test-decimal128-data-type.rb

@@ -1,31 +0,0 @@
-# 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 Decimal128DataTypeTest < Test::Unit::TestCase
-  sub_test_case(".new") do
-    test("ordered arguments") do
-      assert_equal("decimal128(8, 2)",
-                   Arrow::Decimal128DataType.new(8, 2).to_s)
-    end
-
-    test("description") do
-      assert_equal("decimal128(8, 2)",
-                   Arrow::Decimal128DataType.new(precision: 8,
-                                                 scale: 2).to_s)
-    end
-  end
-end

data/test/test-decimal128.rb

@@ -1,102 +0,0 @@
-# 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 Decimal128Test < Test::Unit::TestCase
-  sub_test_case("instance methods") do
-    def setup
-      @decimal128 = Arrow::Decimal128.new("10.1")
-    end
-
-    sub_test_case("#==") do
-      test("Arrow::Decimal128") do
-        assert do
-          @decimal128 == @decimal128
-        end
-      end
-
-      test("not Arrow::Decimal128") do
-        assert do
-          not (@decimal128 == 10.1)
-        end
-      end
-    end
-
-    sub_test_case("#!=") do
-      test("Arrow::Decimal128") do
-        assert do
-          not (@decimal128 != @decimal128)
-        end
-      end
-
-      test("not Arrow::Decimal128") do
-        assert do
-          @decimal128 != 10.1
-        end
-      end
-    end
-
-    sub_test_case("#to_s") do
-      test("default") do
-        assert_equal("101",
-                     @decimal128.to_s)
-      end
-
-      test("scale") do
-        assert_equal("10.1",
-                     @decimal128.to_s(1))
-      end
-    end
-
-    test("#abs") do
-      decimal128 = Arrow::Decimal128.new("-10.1")
-      assert_equal([
-                     Arrow::Decimal128.new("-10.1"),
-                     Arrow::Decimal128.new("10.1"),
-                   ],
-                   [
-                     decimal128,
-                     decimal128.abs,
-                   ])
-    end
-
-    test("#abs!") do
-      decimal128 = Arrow::Decimal128.new("-10.1")
-      decimal128.abs!
-      assert_equal(Arrow::Decimal128.new("10.1"),
-                   decimal128)
-    end
-
-    test("#negate") do
-      decimal128 = Arrow::Decimal128.new("-10.1")
-      assert_equal([
-                     Arrow::Decimal128.new("-10.1"),
-                     Arrow::Decimal128.new("10.1"),
-                   ],
-                   [
-                     decimal128,
-                     decimal128.negate,
-                   ])
-    end
-
-    test("#negate!") do
-      decimal128 = Arrow::Decimal128.new("-10.1")
-      decimal128.negate!
-      assert_equal(Arrow::Decimal128.new("10.1"),
-                   decimal128)
-    end
-  end
-end

data/test/test-decimal256-array-builder.rb

@@ -1,112 +0,0 @@
-# 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 Decimal256ArrayBuilderTest < Test::Unit::TestCase
-  def setup
-    @data_type = Arrow::Decimal256DataType.new(3, 1)
-    @builder = Arrow::Decimal256ArrayBuilder.new(@data_type)
-  end
-
-  sub_test_case("#append_value") do
-    test("nil") do
-      @builder.append_value(nil)
-      array = @builder.finish
-      assert_equal(nil, array[0])
-    end
-
-    test("Arrow::Decimal256") do
-      @builder.append_value(Arrow::Decimal256.new("10.1"))
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-
-    test("String") do
-      @builder.append_value("10.1")
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-
-    test("Float") do
-      @builder.append_value(10.1)
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-
-    test("BigDecimal") do
-      @builder.append_value(BigDecimal("10.1"))
-      array = @builder.finish
-      assert_equal(BigDecimal("10.1"),
-                   array[0])
-    end
-  end
-
-  sub_test_case("#append_values") do
-    test("mixed") do
-      @builder.append_values([
-                               Arrow::Decimal256.new("10.1"),
-                               nil,
-                               "10.1",
-                               10.1,
-                               BigDecimal("10.1"),
-                             ])
-      array = @builder.finish
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                     BigDecimal("10.1"),
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-
-    test("is_valids") do
-      @builder.append_values([
-                               Arrow::Decimal256.new("10.1"),
-                               Arrow::Decimal256.new("10.1"),
-                               Arrow::Decimal256.new("10.1"),
-                             ],
-                             [
-                               true,
-                               false,
-                               true,
-                             ])
-      array = @builder.finish
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-
-    test("packed") do
-      @builder.append_values(Arrow::Decimal256.new("10.1").to_bytes.to_s * 3,
-                             [true, false, true])
-      array = @builder.finish
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-  end
-end

data/test/test-decimal256-array.rb

@@ -1,38 +0,0 @@
-# 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 Decimal256ArrayTest < Test::Unit::TestCase
-  sub_test_case(".new") do
-    test("build") do
-      data_type = Arrow::Decimal256DataType.new(3, 1)
-      values = [
-        10.1,
-        nil,
-        "10.1",
-        BigDecimal("10.1"),
-      ]
-      array = Arrow::Decimal256Array.new(data_type, values)
-      assert_equal([
-                     BigDecimal("10.1"),
-                     nil,
-                     BigDecimal("10.1"),
-                     BigDecimal("10.1"),
-                   ],
-                   array.to_a)
-    end
-  end
-end

data/test/test-decimal256.rb

@@ -1,102 +0,0 @@
-# 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 Decimal256Test < Test::Unit::TestCase
-  sub_test_case("instance methods") do
-    def setup
-      @decimal256 = Arrow::Decimal256.new("10.1")
-    end
-
-    sub_test_case("#==") do
-      test("Arrow::Decimal256") do
-        assert do
-          @decimal256 == @decimal256
-        end
-      end
-
-      test("not Arrow::Decimal256") do
-        assert do
-          not (@decimal256 == 10.1)
-        end
-      end
-    end
-
-    sub_test_case("#!=") do
-      test("Arrow::Decimal256") do
-        assert do
-          not (@decimal256 != @decimal256)
-        end
-      end
-
-      test("not Arrow::Decimal256") do
-        assert do
-          @decimal256 != 10.1
-        end
-      end
-    end
-
-    sub_test_case("#to_s") do
-      test("default") do
-        assert_equal("101",
-                     @decimal256.to_s)
-      end
-
-      test("scale") do
-        assert_equal("10.1",
-                     @decimal256.to_s(1))
-      end
-    end
-
-    test("#abs") do
-      decimal256 = Arrow::Decimal256.new("-10.1")
-      assert_equal([
-                     Arrow::Decimal256.new("-10.1"),
-                     Arrow::Decimal256.new("10.1"),
-                   ],
-                   [
-                     decimal256,
-                     decimal256.abs,
-                   ])
-    end
-
-    test("#abs!") do
-      decimal256 = Arrow::Decimal256.new("-10.1")
-      decimal256.abs!
-      assert_equal(Arrow::Decimal256.new("10.1"),
-                   decimal256)
-    end
-
-    test("#negate") do
-      decimal256 = Arrow::Decimal256.new("-10.1")
-      assert_equal([
-                     Arrow::Decimal256.new("-10.1"),
-                     Arrow::Decimal256.new("10.1"),
-                   ],
-                   [
-                     decimal256,
-                     decimal256.negate,
-                   ])
-    end
-
-    test("#negate!") do
-      decimal256 = Arrow::Decimal256.new("-10.1")
-      decimal256.negate!
-      assert_equal(Arrow::Decimal256.new("10.1"),
-                   decimal256)
-    end
-  end
-end

data/test/test-dense-union-data-type.rb

@@ -1,41 +0,0 @@
-# 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 DenseUnionDataTypeTest < Test::Unit::TestCase
-  sub_test_case(".new") do
-    def setup
-      @fields = [
-        Arrow::Field.new("visible", :boolean),
-        {
-          name: "count",
-          type: :int32,
-        },
-      ]
-    end
-
-    test("ordered arguments") do
-      assert_equal("dense_union<visible: bool=2, count: int32=9>",
-                   Arrow::DenseUnionDataType.new(@fields, [2, 9]).to_s)
-    end
-
-    test("description") do
-      assert_equal("dense_union<visible: bool=2, count: int32=9>",
-                   Arrow::DenseUnionDataType.new(fields: @fields,
-                                                  type_codes: [2, 9]).to_s)
-    end
-  end
-end

data/test/test-dictionary-data-type.rb

@@ -1,40 +0,0 @@
-# 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 DictionaryDataTypeTest < Test::Unit::TestCase
-  sub_test_case(".new") do
-    def setup
-      @index_data_type = :int8
-      @value_data_type = :string
-      @ordered = true
-    end
-
-    test("ordered arguments") do
-      assert_equal("dictionary<values=string, indices=int8, ordered=1>",
-                   Arrow::DictionaryDataType.new(@index_data_type,
-                                                 @value_data_type,
-                                                 @ordered).to_s)
-    end
-
-    test("description") do
-      assert_equal("dictionary<values=string, indices=int8, ordered=1>",
-                   Arrow::DictionaryDataType.new(index_data_type: @index_data_type,
-                                                 value_data_type: @value_data_type,
-                                                 ordered: @ordered).to_s)
-    end
-  end
-end

data/test/test-expression.rb

@@ -1,40 +0,0 @@
-# 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 TestExpression < Test::Unit::TestCase
-  sub_test_case(".try_convert") do
-    test("Symbol") do
-      assert_equal(Arrow::FieldExpression.new("visible"),
-                   Arrow::Expression.try_convert(:visible))
-    end
-
-    test("[String]") do
-      assert_equal(Arrow::CallExpression.new("func", []),
-                   Arrow::Expression.try_convert(["func"]))
-    end
-
-    test("[Symbol]") do
-      assert_equal(Arrow::CallExpression.new("func", []),
-                   Arrow::Expression.try_convert([:func]))
-    end
-
-    test("[String, String]") do
-      assert_equal(Arrow::CallExpression.new("func", ["argument1"]),
-                   Arrow::Expression.try_convert(["func", "argument1"]))
-    end
-  end
-end

data/test/test-feather.rb

@@ -1,49 +0,0 @@
-# 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 FeatherTest < Test::Unit::TestCase
-  include Helper::Fixture
-
-  def setup
-    columns = {
-      "message" => Arrow::StringArray.new(["Start", "Crash", "Shutdown"]),
-      "is_critical" => Arrow::BooleanArray.new([false, true, false]),
-    }
-    @table = Arrow::Table.new(columns)
-
-    @output = Tempfile.new(["red-arrow", ".feather"])
-    begin
-      yield(@output)
-    ensure
-      @output.close!
-    end
-  end
-
-  def test_default
-    @table.save(@output.path)
-    @output.close
-
-    assert_equal(@table, Arrow::Table.load(@output.path))
-  end
-
-  def test_compression
-    @table.save(@output.path, compression: :zstd)
-    @output.close
-
-    assert_equal(@table, Arrow::Table.load(@output.path))
-  end
-end