red-arrow 3.0.0 → 6.0.0

data/test/test-slicer.rb

@@ -45,15 +45,15 @@ class SlicerTest < Test::Unit::TestCase
         slicer.visible
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    1	true   
-2	     	       
-3	    8	true   
-4	   16	true   
-5	     	       
-6	     	       
-7	  256	true   
+	 count	visible
+0	(null)	 (null)
+1	     1	true   
+2	(null)	 (null)
+3	     8	true   
+4	    16	true   
+5	(null)	 (null)
+6	(null)	 (null)
+7	   256	true   
       TABLE
     end
 
@@ -62,16 +62,16 @@ class SlicerTest < Test::Unit::TestCase
         slicer.count
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    1	true   
-1	    2	false  
-2	    4	       
-3	    8	true   
-4	   16	true   
-5	   32	false  
-6	   64	       
-7	     	       
-8	  256	true   
+	 count	visible
+0	     1	true   
+1	     2	false  
+2	     4	 (null)
+3	     8	true   
+4	    16	true   
+5	    32	false  
+6	    64	 (null)
+7	(null)	 (null)
+8	   256	true   
       TABLE
     end
   end
@@ -82,13 +82,13 @@ class SlicerTest < Test::Unit::TestCase
         !slicer.visible
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    2	false  
-2	     	       
-3	   32	false  
-4	     	       
-5	     	       
+	 count	visible
+0	(null)	 (null)
+1	     2	false  
+2	(null)	 (null)
+3	    32	false  
+4	(null)	 (null)
+5	(null)	 (null)
       TABLE
     end
 
@@ -97,9 +97,9 @@ class SlicerTest < Test::Unit::TestCase
         !slicer.count
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	     	       
+	 count	visible
+0	     0	 (null)
+1	(null)	 (null)
       TABLE
     end
   end
@@ -109,11 +109,11 @@ class SlicerTest < Test::Unit::TestCase
       slicer.visible.null?
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    4	       
-2	   64	       
-3	     	       
+	 count	visible
+0	     0	 (null)
+1	     4	 (null)
+2	    64	 (null)
+3	(null)	 (null)
     TABLE
   end
 
@@ -138,11 +138,11 @@ class SlicerTest < Test::Unit::TestCase
         slicer.visible == nil
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    4	       
-2	   64	       
-3	     	       
+	 count	visible
+0	     0	 (null)
+1	     4	 (null)
+2	    64	 (null)
+3	(null)	 (null)
       TABLE
     end
 
@@ -151,15 +151,15 @@ class SlicerTest < Test::Unit::TestCase
         slicer.visible == true
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    1	true   
-2	     	       
-3	    8	true   
-4	   16	true   
-5	     	       
-6	     	       
-7	  256	true   
+	 count	visible
+0	(null)	 (null)
+1	     1	true   
+2	(null)	 (null)
+3	     8	true   
+4	    16	true   
+5	(null)	 (null)
+6	(null)	 (null)
+7	   256	true   
       TABLE
     end
   end
@@ -185,13 +185,13 @@ class SlicerTest < Test::Unit::TestCase
         !(slicer.visible == true)
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    2	false  
-2	     	       
-3	   32	false  
-4	     	       
-5	     	       
+	 count	visible
+0	(null)	 (null)
+1	     2	false  
+2	(null)	 (null)
+3	    32	false  
+4	(null)	 (null)
+5	(null)	 (null)
       TABLE
     end
   end
@@ -217,13 +217,13 @@ class SlicerTest < Test::Unit::TestCase
         slicer.visible != true
       end
       assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    2	false  
-2	     	       
-3	   32	false  
-4	     	       
-5	     	       
+	 count	visible
+0	(null)	 (null)
+1	     2	false  
+2	(null)	 (null)
+3	    32	false  
+4	(null)	 (null)
+5	(null)	 (null)
       TABLE
     end
   end
@@ -233,13 +233,13 @@ class SlicerTest < Test::Unit::TestCase
       slicer.count < 16
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    1	true   
-2	    2	false  
-3	    4	       
-4	    8	true   
-5	     	       
+	 count	visible
+0	     0	 (null)
+1	     1	true   
+2	     2	false  
+3	     4	 (null)
+4	     8	true   
+5	(null)	 (null)
     TABLE
   end
 
@@ -248,12 +248,12 @@ class SlicerTest < Test::Unit::TestCase
       !(slicer.count < 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	   16	true   
-1	   32	false  
-2	   64	       
-3	     	       
-4	  256	true   
+	 count	visible
+0	    16	true   
+1	    32	false  
+2	    64	 (null)
+3	(null)	 (null)
+4	   256	true   
     TABLE
   end
 
@@ -262,14 +262,14 @@ class SlicerTest < Test::Unit::TestCase
       slicer.count <= 16
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    1	true   
-2	    2	false  
-3	    4	       
-4	    8	true   
-5	   16	true   
-6	     	       
+	 count	visible
+0	     0	 (null)
+1	     1	true   
+2	     2	false  
+3	     4	 (null)
+4	     8	true   
+5	    16	true   
+6	(null)	 (null)
     TABLE
   end
 
@@ -278,11 +278,11 @@ class SlicerTest < Test::Unit::TestCase
       !(slicer.count <= 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	   32	false  
-1	   64	       
-2	     	       
-3	  256	true   
+	 count	visible
+0	    32	false  
+1	    64	 (null)
+2	(null)	 (null)
+3	   256	true   
     TABLE
   end
 
@@ -291,11 +291,11 @@ class SlicerTest < Test::Unit::TestCase
       slicer.count > 16
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	   32	false  
-1	   64	       
-2	     	       
-3	  256	true   
+	 count	visible
+0	    32	false  
+1	    64	 (null)
+2	(null)	 (null)
+3	   256	true   
     TABLE
   end
 
@@ -304,14 +304,14 @@ class SlicerTest < Test::Unit::TestCase
       !(slicer.count > 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    1	true   
-2	    2	false  
-3	    4	       
-4	    8	true   
-5	   16	true   
-6	     	       
+	 count	visible
+0	     0	 (null)
+1	     1	true   
+2	     2	false  
+3	     4	 (null)
+4	     8	true   
+5	    16	true   
+6	(null)	 (null)
     TABLE
   end
 
@@ -320,12 +320,12 @@ class SlicerTest < Test::Unit::TestCase
       slicer.count >= 16
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	   16	true   
-1	   32	false  
-2	   64	       
-3	     	       
-4	  256	true   
+	 count	visible
+0	    16	true   
+1	    32	false  
+2	    64	 (null)
+3	(null)	 (null)
+4	   256	true   
     TABLE
   end
 
@@ -334,13 +334,13 @@ class SlicerTest < Test::Unit::TestCase
       !(slicer.count >= 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    1	true   
-2	    2	false  
-3	    4	       
-4	    8	true   
-5	     	       
+	 count	visible
+0	     0	 (null)
+1	     1	true   
+2	     2	false  
+3	     4	 (null)
+4	     8	true   
+5	(null)	 (null)
     TABLE
   end
 
@@ -351,10 +351,9 @@ class SlicerTest < Test::Unit::TestCase
     assert_equal(<<-TABLE, sliced_table.to_s)
 	count	visible
 0	    1	true   
-1	    4	       
+1	    4	 (null)
 2	   16	true   
-3	   64	       
-4	     	       
+3	   64	 (null)
     TABLE
   end
 
@@ -363,13 +362,13 @@ class SlicerTest < Test::Unit::TestCase
       !slicer.count.in?([1, 4, 16, 64])
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    2	false  
-2	    8	true   
-3	   32	false  
-4	     	       
-5	  256	true   
+	 count	visible
+0	     0	 (null)
+1	     2	false  
+2	     8	true   
+3	    32	false  
+4	(null)	 (null)
+5	   256	true   
     TABLE
   end
 
@@ -378,13 +377,13 @@ class SlicerTest < Test::Unit::TestCase
       slicer.visible & (slicer.count >= 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	     	       
-2	   16	true   
-3	     	       
-4	     	       
-5	  256	true   
+	 count	visible
+0	(null)	 (null)
+1	(null)	 (null)
+2	    16	true   
+3	(null)	 (null)
+4	(null)	 (null)
+5	   256	true   
     TABLE
   end
 
@@ -393,16 +392,16 @@ class SlicerTest < Test::Unit::TestCase
       slicer.visible | (slicer.count >= 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    1	true   
-2	     	       
-3	    8	true   
-4	   16	true   
-5	   32	false  
-6	     	       
-7	     	       
-8	  256	true   
+	 count	visible
+0	(null)	 (null)
+1	     1	true   
+2	(null)	 (null)
+3	     8	true   
+4	    16	true   
+5	    32	false  
+6	(null)	 (null)
+7	(null)	 (null)
+8	   256	true   
     TABLE
   end
 
@@ -411,14 +410,14 @@ class SlicerTest < Test::Unit::TestCase
       slicer.visible ^ (slicer.count >= 16)
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	     	       
-1	    1	true   
-2	     	       
-3	    8	true   
-4	   32	false  
-5	     	       
-6	     	       
+	 count	visible
+0	(null)	 (null)
+1	     1	true   
+2	(null)	 (null)
+3	     8	true   
+4	    32	false  
+5	(null)	 (null)
+6	(null)	 (null)
     TABLE
   end
 
@@ -429,15 +428,15 @@ class SlicerTest < Test::Unit::TestCase
       end
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    1	true   
-2	    4	       
-3	    8	true   
-4	   16	true   
-5	   64	       
-6	     	       
-7	  256	true   
+	 count	visible
+0	     0	 (null)
+1	     1	true   
+2	     4	 (null)
+3	     8	true   
+4	    16	true   
+5	    64	 (null)
+6	(null)	 (null)
+7	   256	true   
     TABLE
   end
 
@@ -474,15 +473,15 @@ class SlicerTest < Test::Unit::TestCase
       end
     end
     assert_equal(<<-TABLE, sliced_table.to_s)
-	count	visible
-0	    0	       
-1	    1	true   
-2	    4	       
-3	    8	true   
-4	   16	true   
-5	   64	       
-6	     	       
-7	  256	true   
+	 count	visible
+0	     0	 (null)
+1	     1	true   
+2	     4	 (null)
+3	     8	true   
+4	    16	true   
+5	    64	 (null)
+6	(null)	 (null)
+7	   256	true   
     TABLE
   end
 end

data/test/test-sort-indices.rb

@@ -0,0 +1,40 @@
+# 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 SortIndicesTest < Test::Unit::TestCase
+  def setup
+    @table = Arrow::Table.new(number1: [16, -1,   2, 32, -4, -4, -8],
+                              number2: [32,  2, -16,  8,  1,  4,  1])
+  end
+
+  sub_test_case("Table") do
+    test("Symbol") do
+      assert_equal(Arrow::UInt64Array.new([6, 4, 5, 1, 2, 0, 3]),
+                   @table.sort_indices(:number1))
+    end
+
+    test("-String") do
+      assert_equal(Arrow::UInt64Array.new([3, 0, 2, 1, 4, 5, 6]),
+                   @table.sort_indices("-number1"))
+    end
+
+    test("Symbol, -String") do
+      assert_equal(Arrow::UInt64Array.new([6, 5, 4, 1, 2, 0, 3]),
+                   @table.sort_indices([:number1, "-number2"]))
+    end
+  end
+end

data/test/test-sort-key.rb

@@ -0,0 +1,81 @@
+# 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 SortKeyTest < Test::Unit::TestCase
+  sub_test_case(".resolve") do
+    test("SortKey") do
+      assert_equal(Arrow::SortKey.new("-count"),
+                   Arrow::SortKey.resolve(Arrow::SortKey.new("-count")))
+    end
+
+    test("-String") do
+      assert_equal(Arrow::SortKey.new("-count"),
+                   Arrow::SortKey.resolve("-count"))
+    end
+
+    test("Symbol, Symbol") do
+      assert_equal(Arrow::SortKey.new("-count"),
+                   Arrow::SortKey.resolve(:count, :desc))
+    end
+  end
+
+  sub_test_case("#initialize") do
+    test("String") do
+      assert_equal("+count",
+                   Arrow::SortKey.new("count").to_s)
+    end
+
+    test("+String") do
+      assert_equal("+count",
+                   Arrow::SortKey.new("+count").to_s)
+    end
+
+    test("-String") do
+      assert_equal("-count",
+                   Arrow::SortKey.new("-count").to_s)
+    end
+
+    test("Symbol") do
+      assert_equal("+-count",
+                   Arrow::SortKey.new(:"-count").to_s)
+    end
+
+    test("String, Symbol") do
+      assert_equal("--count",
+                   Arrow::SortKey.new("-count", :desc).to_s)
+    end
+
+    test("String, String") do
+      assert_equal("--count",
+                   Arrow::SortKey.new("-count", "desc").to_s)
+    end
+
+    test("String, SortOrder") do
+      assert_equal("--count",
+                   Arrow::SortKey.new("-count",
+                                      Arrow::SortOrder::DESCENDING).to_s)
+    end
+  end
+
+  sub_test_case("#to_s") do
+    test("recreatable") do
+      key = Arrow::SortKey.new("-count", :desc)
+      assert_equal(key,
+                   Arrow::SortKey.new(key.to_s))
+    end
+  end
+end

data/test/test-sort-options.rb

@@ -0,0 +1,58 @@
+# 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 SortOptionsTest < Test::Unit::TestCase
+  sub_test_case("#initialize") do
+    test("none") do
+      options = Arrow::SortOptions.new
+      assert_equal([],
+                   options.sort_keys.collect(&:to_s))
+    end
+
+    test("-String, Symbol") do
+      options = Arrow::SortOptions.new("-count", :age)
+      assert_equal(["-count", "+age"],
+                   options.sort_keys.collect(&:to_s))
+    end
+  end
+
+  sub_test_case("instance methods") do
+    setup do
+      @options = Arrow::SortOptions.new
+    end
+
+    sub_test_case("#add_sort_key") do
+      test("-String") do
+        @options.add_sort_key("-count")
+        assert_equal(["-count"],
+                     @options.sort_keys.collect(&:to_s))
+      end
+
+      test("-String, Symbol") do
+        @options.add_sort_key("-count", :desc)
+        assert_equal(["--count"],
+                     @options.sort_keys.collect(&:to_s))
+      end
+
+      test("SortKey") do
+        @options.add_sort_key(Arrow::SortKey.new("-count"))
+        assert_equal(["-count"],
+                     @options.sort_keys.collect(&:to_s))
+      end
+    end
+  end
+end