red-arrow 18.1.0 → 19.0.0

data/test/test-decimal256.rb

@@ -1,126 +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("String") do
-        assert do
-          @decimal256 == "10.1"
-        end
-      end
-
-      test("Float") do
-        assert do
-          @decimal256 == 10.1
-        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("String") do
-        assert do
-          not (@decimal256 != "10.1")
-        end
-      end
-
-      test("Float") do
-        assert do
-          not (@decimal256 != 10.1)
-        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-array.rb

@@ -1,42 +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 DenseUnionArrayTest < Test::Unit::TestCase
-  def setup
-    data_type_fields = [
-      Arrow::Field.new("number", :int16),
-      Arrow::Field.new("text", :string),
-    ]
-    type_codes = [11, 13]
-    @data_type = Arrow::DenseUnionDataType.new(data_type_fields, type_codes)
-    type_ids = Arrow::Int8Array.new([11, 13, 11, 13, 13])
-    value_offsets = Arrow::Int32Array.new([0, 0, 1, 1, 2])
-    fields = [
-      Arrow::Int16Array.new([1, nil]),
-      Arrow::StringArray.new(["a", "b", "c"])
-    ]
-    @array = Arrow::DenseUnionArray.new(@data_type,
-                                        type_ids,
-                                        value_offsets,
-                                        fields)
-  end
-
-  def test_get_value
-    assert_equal([1, "a", nil, "b", "c"],
-                 @array.length.times.collect {|i| @array[i]})
-  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-array.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 DictionaryArrayTest < Test::Unit::TestCase
-  sub_test_case("instance methods") do
-    def setup
-      @values = ["a", "b", "c", "b", "a"]
-      @string_array = Arrow::StringArray.new(@values)
-      @array = @string_array.dictionary_encode
-    end
-
-    test("#[]") do
-      assert_equal(@values, @array.to_a)
-    end
-
-    test("#get_value") do
-      assert_equal([
-                     @values[0],
-                     @values[3],
-                   ],
-                   [
-                     @array.get_value(0),
-                     @array.get_value(3),
-                   ])
-    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,51 +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
-
-    test("[Symbol, String, Hash]") do
-      options = Arrow::MatchSubstringOptions.new
-      options.pattern = "hello"
-      assert_equal(Arrow::CallExpression.new("match_substring",
-                                             ["content"],
-                                             options),
-                   Arrow::Expression.try_convert([:match_substring,
-                                                  "content",
-                                                  {pattern: "hello"}]))
-    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

data/test/test-field.rb

@@ -1,117 +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 FieldTest < Test::Unit::TestCase
-  sub_test_case(".new") do
-    test("String, Arrow::DataType") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new("visible", Arrow::BooleanDataType.new).to_s)
-    end
-
-    test("String, Arrow::DataType, Boolean") do
-      assert_equal("visible: bool not null",
-                   Arrow::Field.new("visible",
-                                    Arrow::BooleanDataType.new,
-                                    false).to_s)
-    end
-
-    test("Symbol, Arrow::DataType") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new(:visible, Arrow::BooleanDataType.new).to_s)
-    end
-
-    test("String, Symbol") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new(:visible, :boolean).to_s)
-    end
-
-    test("String, Hash") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new(:visible, type: :boolean).to_s)
-    end
-
-    test("String, Hash (:nullable)") do
-      assert_equal("visible: bool not null",
-                   Arrow::Field.new(:visible,
-                                    type: :boolean,
-                                    nullable: false).to_s)
-    end
-
-    test("String, Hash, Boolean)") do
-      assert_equal("visible: bool not null",
-                   Arrow::Field.new(:visible, {type: :boolean}, false).to_s)
-    end
-
-    test("description: String") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new(name: "visible",
-                                    data_type: :boolean).to_s)
-    end
-
-    test("description: Symbol") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new(name: :visible,
-                                    data_type: :boolean).to_s)
-    end
-
-    test("description: shortcut") do
-      assert_equal("visible: bool",
-                   Arrow::Field.new(name: :visible,
-                                    type: :boolean).to_s)
-    end
-
-    test("description: nullable") do
-      assert_equal("visible: bool not null",
-                   Arrow::Field.new(name: :visible,
-                                    data_type: :boolean,
-                                    nullable: false).to_s)
-    end
-
-    test("Hash: shortcut: additional") do
-      description = {
-        name: :tags,
-        type: :list,
-        field: {
-          name: "tag",
-          type: :string,
-        },
-      }
-      assert_equal("tags: list<tag: string>",
-                   Arrow::Field.new(description).to_s)
-    end
-  end
-
-  sub_test_case("instance methods") do
-    def setup
-      @field = Arrow::Field.new("count", :uint32)
-    end
-
-    sub_test_case("#==") do
-      test("Arrow::Field") do
-        assert do
-          @field == @field
-        end
-      end
-
-      test("not Arrow::Field") do
-        assert do
-          not (@field == 29)
-        end
-      end
-    end
-  end
-end

data/test/test-file-output-stream.rb

@@ -1,54 +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 TestFileOutputStream < Test::Unit::TestCase
-  sub_test_case(".open") do
-    def setup
-      @file = Tempfile.open("arrow-file-output-stream")
-      @file.write("Hello")
-      @file.close
-    end
-
-    def test_default
-      Arrow::FileOutputStream.open(@file.path) do |file|
-        file.write(" World")
-      end
-      assert_equal(" World", File.read(@file.path))
-    end
-
-    def test_options_append
-      Arrow::FileOutputStream.open(@file.path, append: true) do |file|
-        file.write(" World")
-      end
-      assert_equal("Hello World", File.read(@file.path))
-    end
-
-    def test_append_true
-      Arrow::FileOutputStream.open(@file.path, true) do |file|
-        file.write(" World")
-      end
-      assert_equal("Hello World", File.read(@file.path))
-    end
-
-    def test_append_false
-      Arrow::FileOutputStream.open(@file.path, false) do |file|
-        file.write(" World")
-      end
-      assert_equal(" World", File.read(@file.path))
-    end
-  end
-end

data/test/test-fixed-size-binary-array-builder.rb

@@ -1,92 +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 FixedSizeBinaryArrayBuilderTest < Test::Unit::TestCase
-  def setup
-    @data_type = Arrow::FixedSizeBinaryDataType.new(4)
-    @builder = Arrow::FixedSizeBinaryArrayBuilder.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("String") do
-      @builder.append_value("0123")
-      array = @builder.finish
-      assert_equal("0123", array[0])
-    end
-
-    test("GLib::Bytes") do
-      @builder.append_value(GLib::Bytes.new("0123"))
-      array = @builder.finish
-      assert_equal("0123", array[0])
-    end
-  end
-
-  sub_test_case("#append_values") do
-    test("mixed") do
-      @builder.append_values([
-                               "0123",
-                               nil,
-                               GLib::Bytes.new("abcd"),
-                             ])
-      array = @builder.finish
-      assert_equal([
-                     "0123",
-                     nil,
-                     "abcd",
-                   ],
-                   array.to_a)
-    end
-
-    test("is_valids") do
-      @builder.append_values([
-                               "0123",
-                               "0123",
-                               "0123",
-                             ],
-                             [
-                               true,
-                               false,
-                               true,
-                             ])
-      array = @builder.finish
-      assert_equal([
-                     "0123",
-                     nil,
-                     "0123",
-                   ],
-                   array.to_a)
-    end
-
-    test("packed") do
-      @builder.append_values("0123" * 3,
-                             [true, false, true])
-      array = @builder.finish
-      assert_equal([
-                     "0123",
-                     nil,
-                     "0123",
-                   ],
-                   array.to_a)
-    end
-  end
-end

data/test/test-fixed-size-binary-array.rb

@@ -1,36 +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 FixedSizeBinaryArrayTest < Test::Unit::TestCase
-  sub_test_case(".new") do
-    test("build") do
-      data_type = Arrow::FixedSizeBinaryDataType.new(4)
-      values = [
-        "0123",
-        nil,
-        GLib::Bytes.new("abcd"),
-      ]
-      array = Arrow::FixedSizeBinaryArray.new(data_type, values)
-      assert_equal([
-                     "0123",
-                     nil,
-                     "abcd",
-                   ],
-                   array.to_a)
-    end
-  end
-end