red-arrow 0.12.0 → 0.13.0

data/test/test-array.rb

@@ -24,15 +24,44 @@ class ArrayTest < Test::Unit::TestCase
     end
   end
 
-  test("#each") do
-    array = Arrow::BooleanArray.new([true, false, nil, true])
-    assert_equal([true, false, nil, true],
-                 array.to_a)
-  end
+  sub_test_case("instance methods") do
+    def setup
+      @values = [true, false, nil, true]
+      @array = Arrow::BooleanArray.new(@values)
+    end
+
+    test("#each") do
+      assert_equal(@values, @array.to_a)
+    end
+
+    sub_test_case("#[]") do
+      test("valid range") do
+        assert_equal(@values,
+                     @array.length.times.collect {|i| @array[i]})
+      end
 
-  test("#[]") do
-    array = Arrow::BooleanArray.new([true, false, nil, true])
-    assert_equal([true, false, nil, true],
-                 [array[0], array[1], array[2], array[3]])
+      test("out of range") do
+        assert_nil(@array[@array.length])
+      end
+
+      test("negative index") do
+        assert_equal(@values.last,
+                     @array[-1])
+      end
+    end
+
+    sub_test_case("#==") do
+      test("Arrow::Array") do
+        assert do
+          @array == @array
+        end
+      end
+
+      test("not Arrow::Array") do
+        assert do
+          not (@array == 29)
+        end
+      end
+    end
   end
 end

data/test/test-bigdecimal.rb

@@ -0,0 +1,23 @@
+# 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 BigDecimalTest < Test::Unit::TestCase
+  test("#to_arrow") do
+    assert_equal(Arrow::Decimal128.new("3.14"),
+                 BigDecimal("3.14").to_arrow)
+  end
+end

data/{dependency-check/Rakefile → test/test-buffer.rb}

@@ -1,5 +1,3 @@
-# -*- 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
@@ -17,26 +15,23 @@
 # specific language governing permissions and limitations
 # under the License.
 
-require "pkg-config"
-require "native-package-installer"
-
-case RUBY_PLATFORM
-when /mingw|mswin/
-  task :default => "nothing"
-else
-  task :default => "dependency:check"
-end
+class BufferTest < Test::Unit::TestCase
+  sub_test_case("instance methods") do
+    def setup
+      @buffer = Arrow::Buffer.new("Hello")
+    end
 
-task :nothing do
-end
+    sub_test_case("#==") do
+      test("Arrow::Buffer") do
+        assert do
+          @buffer == @buffer
+        end
+      end
 
-namespace :dependency do
-  desc "Check dependency"
-  task :check do
-    unless PKGConfig.check_version?("arrow-glib", 0, 9, 0)
-      unless NativePackageInstaller.install(:debian => "libarrow-glib-dev",
-                                            :redhat => "arrow-glib-devel")
-        exit(false)
+      test("not Arrow::Buffer") do
+        assert do
+          not (@buffer == 29)
+        end
       end
     end
   end

data/test/test-chunked-array.rb

@@ -62,4 +62,26 @@ class ChunkedArrayTest < Test::Unit::TestCase
                    ])
     end
   end
+
+  sub_test_case("#==") do
+    def setup
+      arrays = [
+        Arrow::BooleanArray.new([true]),
+        Arrow::BooleanArray.new([false, true]),
+      ]
+      @chunked_array = Arrow::ChunkedArray.new(arrays)
+    end
+
+    test("Arrow::ChunkedArray") do
+      assert do
+        @chunked_array == @chunked_array
+      end
+    end
+
+    test("not Arrow::ChunkedArray") do
+      assert do
+        not (@chunked_array == 29)
+      end
+    end
+  end
 end

data/test/test-column.rb

@@ -40,4 +40,28 @@ class ColumnTest < Test::Unit::TestCase
     assert_equal([1, [true, false, nil, true]],
                  [packed_column.data.n_chunks, packed_column.to_a])
   end
+
+  sub_test_case("#==") do
+    def setup
+      arrays = [
+        Arrow::BooleanArray.new([true]),
+        Arrow::BooleanArray.new([false, true]),
+      ]
+      chunked_array = Arrow::ChunkedArray.new(arrays)
+      @column = Arrow::Column.new(Arrow::Field.new("visible", :boolean),
+                                  chunked_array)
+    end
+
+    test("Arrow::Column") do
+      assert do
+        @column == @column
+      end
+    end
+
+    test("not Arrow::Column") do
+      assert do
+        not (@column == 29)
+      end
+    end
+  end
 end

data/test/test-csv-loader.rb

@@ -141,5 +141,35 @@ count
 4
                    CSV
     end
+
+    test(":encoding") do
+      messages = [
+        "\u3042", # U+3042 HIRAGANA LETTER A
+        "\u3044", # U+3044 HIRAGANA LETTER I
+        "\u3046", # U+3046 HIRAGANA LETTER U
+      ]
+      table = Arrow::Table.new(:message => Arrow::StringArray.new(messages))
+      encoding = "cp932"
+      assert_equal(table,
+                   load_csv((["message"] + messages).join("\n").encode(encoding),
+                            schema: table.schema,
+                            encoding: encoding))
+    end
+
+    test(":encoding and :compression") do
+      messages = [
+        "\u3042", # U+3042 HIRAGANA LETTER A
+        "\u3044", # U+3044 HIRAGANA LETTER I
+        "\u3046", # U+3046 HIRAGANA LETTER U
+      ]
+      table = Arrow::Table.new(:message => Arrow::StringArray.new(messages))
+      encoding = "cp932"
+      csv = (["message"] + messages).join("\n").encode(encoding)
+      assert_equal(table,
+                   load_csv(Zlib::Deflate.deflate(csv),
+                            schema: table.schema,
+                            encoding: encoding,
+                            compression: :gzip))
+    end
   end
 end

data/test/test-data-type.rb

@@ -43,5 +43,30 @@ class DataTypeTest < Test::Unit::TestCase
       assert_equal(Arrow::ListDataType.new(field),
                    Arrow::DataType.resolve(type: :list, field: field))
     end
+
+    test("_") do
+      assert_equal(Arrow::FixedSizeBinaryDataType.new(10),
+                   Arrow::DataType.resolve([:fixed_size_binary, 10]))
+    end
+  end
+
+  sub_test_case("instance methods") do
+    def setup
+      @data_type = Arrow::StringDataType.new
+    end
+
+    sub_test_case("#==") do
+      test("Arrow::DataType") do
+        assert do
+          @data_type == @data_type
+        end
+      end
+
+      test("not Arrow::DataType") do
+        assert do
+          not (@data_type == 29)
+        end
+      end
+    end
   end
 end

data/test/test-decimal128.rb

@@ -0,0 +1,64 @@
+# 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
+  end
+end

data/test/test-field.rb

@@ -68,4 +68,24 @@ class FieldTest < Test::Unit::TestCase
                    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-group.rb

@@ -43,8 +43,8 @@ class GroupTest < Test::Unit::TestCase
       table = Arrow::Table.new(raw_table)
       assert_equal(<<-TABLE, table.group(:time).count.to_s)
 	                     time	int
-0	2018-01-29T00:00:00+09:00	  1
-1	2018-01-30T00:00:00+09:00	  1
+0	#{time_values[0].iso8601}	  1
+1	#{time_values[1].iso8601}	  1
       TABLE
     end
   end

data/test/test-record-batch-builder.rb

@@ -112,5 +112,14 @@ class RecordBatchBuilderTest < Test::Unit::TestCase
                                           arrays),
                    @builder.flush)
     end
+
+    test("#column_builders") do
+      column_builders = [
+        @builder.get_column_builder(0),
+        @builder.get_column_builder(1),
+      ]
+      assert_equal(column_builders,
+                   @builder.column_builders)
+    end
   end
 end

data/test/test-record-batch.rb

@@ -108,5 +108,19 @@ class RecordBatchTest < Test::Unit::TestCase
       assert_equal(Arrow::Table.new(@schema, [@counts]),
                    @record_batch.to_table)
     end
+
+    sub_test_case("#==") do
+      test("Arrow::RecordBatch") do
+        assert do
+          @record_batch == @record_batch
+        end
+      end
+
+      test("not Arrow::RecordBatch") do
+        assert do
+          not (@record_batch == 29)
+        end
+      end
+    end
   end
 end

data/test/test-schema.rb

@@ -100,5 +100,19 @@ class SchemaTest < Test::Unit::TestCase
         end
       end
     end
+
+    sub_test_case("#==") do
+      test("Arrow::Schema") do
+        assert do
+          @schema == @schema
+        end
+      end
+
+      test("not Arrow::Schema") do
+        assert do
+          not (@schema == 29)
+        end
+      end
+    end
   end
 end

data/test/test-struct-array.rb

@@ -49,9 +49,22 @@ class StructArrayTest < Test::Unit::TestCase
     end
 
     test("#[]") do
-      notify("TODO: Returns Arrow::Struct instead.")
-      assert_equal([[true, false], [1, 2]],
-                   [@array[0].to_a, @array[1].to_a])
+      assert_equal([
+                     Arrow::Struct.new(@array, 0),
+                     Arrow::Struct.new(@array, 1),
+                   ],
+                   @array.to_a)
+    end
+
+    test("#get_value") do
+      assert_equal([
+                     Arrow::Struct.new(@array, 0),
+                     Arrow::Struct.new(@array, 1),
+                   ],
+                   [
+                     @array.get_value(0),
+                     @array.get_value(1),
+                   ])
     end
 
     sub_test_case("#find_field") do

data/test/test-table.rb

@@ -596,5 +596,19 @@ visible: false
         end
       end
     end
+
+    sub_test_case("#==") do
+      test("Arrow::Table") do
+        assert do
+          @table == @table
+        end
+      end
+
+      test("not Arrow::Table") do
+        assert do
+          not (@table == 29)
+        end
+      end
+    end
   end
 end