carray 1.1.4 → 1.1.5

data/test/test_math.rb

@@ -0,0 +1,98 @@
+$:.unshift("../lib", "..")
+
+require 'test/unit'
+require 'carray'
+
+class TestMath < Test::Unit::TestCase
+
+  def test_zerodiv
+    zero = CArray.int(3,3) { 0 }
+    one  = CArray.int(3,3) { 1 }
+    assert_raise(ZeroDivisionError) { 1/zero }
+    assert_raise(ZeroDivisionError) { one/zero } # div, div!
+    assert_raise(ZeroDivisionError) { one/0 }
+    assert_raise(ZeroDivisionError) { one/zero }
+    assert_raise(ZeroDivisionError) { one % 0 }  # mod, mod!
+    assert_raise(ZeroDivisionError) { one % zero }
+    assert_raise(ZeroDivisionError) { zero.rcp } # rcp!
+    assert_raise(ZeroDivisionError) { zero.rcp_mul(1) } # rcp_mul!
+    assert_raise(ZeroDivisionError) { zero.rcp_mul(one) }
+    assert_raise(ZeroDivisionError) { 0 ** (-one) } # pow, pow!
+    assert_raise(ZeroDivisionError) { zero ** (-one) }
+  end
+
+  def test_cmp
+    a = CA_INT([0,1,2,3,4])
+    b = CA_INT([4,3,2,1,0])
+
+    # ---
+    assert_equal(CA_INT8([-1,-1,0,1,1]), a <=> 2)
+    assert_equal(CA_INT8([1,1,0,-1,-1]), 2 <=> a)
+    assert_equal(CA_INT8([-1,-1,0,1,1]), a <=> b)
+  end
+
+  def test_bit_op
+    a = CArray.int(10).seq!
+
+    # ---
+    assert_equal(a.convert{|x| ~x}, ~a)
+
+    assert_equal(a.convert{|x| x & 1}, a & 1)
+    assert_equal(a & 1, 1 & a)
+
+    assert_equal(a.convert{|x| x | 1}, a | 1)
+    assert_equal(a | 1, 1 | a)
+
+    assert_equal(a.convert{|x| x ^ 1}, a ^ 1)
+    assert_equal(a ^ 1, 1 ^ a)
+
+    # ---
+    assert_equal(a.convert{|x| x << 1}, a << 1)
+    assert_equal(a.convert{|x| 1 << x}, 1 << a)
+
+    assert_equal(a.convert{|x| x >> 1}, a >> 1)
+    assert_equal(a.convert{|x| 1 >> x}, 1 >> a)
+  end
+
+  def test_max_min
+    a = CArray.int(3,3).seq!
+    b = a.reverse
+
+    # ---
+    assert_equal(CA_INT32([[8,7,6],
+                           [5,4,5],
+                           [6,7,8]]), a.pmax(b))
+    assert_equal(CA_INT32([[0,1,2],
+                           [3,4,3],
+                           [2,1,0]]), a.pmin(b))
+    # ---
+    assert_equal(CA_INT32([[0,0,0],
+                           [0,0,0],
+                           [0,0,0]]), a.pmin(0))
+    assert_equal(CA_INT32([[8,8,8],
+                           [8,8,8],
+                           [8,8,8]]), a.pmax(8))
+    # ---
+    assert_equal(CA_INT32([[0,0,0],
+                           [0,0,0],
+                           [0,0,0]]), CA_INT32(0).pmin(a))
+    assert_equal(CA_INT32([[8,8,8],
+                           [8,8,8],
+                           [8,8,8]]), CA_INT32(8).pmax(a))
+    # ---
+    assert_equal(CA_INT32([[8,7,6],
+                           [5,4,5],
+                           [6,7,8]]), CAMath.max(a, b))
+    assert_equal(CA_INT32([[0,1,2],
+                           [3,4,3],
+                           [2,1,0]]), CAMath.min(a, b))
+    # ---
+    assert_equal(CA_INT32([[0,0,0],
+                           [0,0,0],
+                           [0,0,0]]), CAMath.min(0,a,b,8))
+    assert_equal(CA_INT32([[8,8,8],
+                           [8,8,8],
+                           [8,8,8]]), CAMath.max(0,a,b,8))
+  end
+
+end

data/test/test_narray.rb

@@ -0,0 +1,64 @@
+$:.unshift("../lib", "..")
+
+require 'test/unit'
+require 'carray'
+
+class TestCArrayNArray < Test::Unit::TestCase
+
+  def test_na
+    a = CArray.int32(3, 3).seq!
+    assert_equal(NArray.int(3,3).indgen!, a.na)
+    assert_equal(NArray.int(9).indgen!, a.na(9))
+
+    b = a.na
+    assert_equal(a, b.ca)
+    assert_equal(CArray.int(9).seq!, b.ca(9))
+  end
+
+  def test_to_na
+    a = CArray.int32(3, 3).seq!
+    assert_equal(NArray.int(3,3).indgen!, a.to_na)
+
+    b = a.to_na
+    assert_equal(a, b.to_ca)
+  end
+
+  def test_asign
+    a = CArray.int32(3, 3) { 0 }
+    b = NArray.int(3,3).indgen!
+    a[] = b
+    assert_equal(b.ca, a)
+  end
+
+  def test_empty_array
+
+    na0  = NArray.int(0)
+    ca0  = CArray.int(0)
+    ca30 = CArray.int(3,0)
+
+    #--- na
+    assert_equal(na0, ca0.na)
+    assert_equal(na0, ca30.na)
+
+    #--- to_na
+    assert_equal(na0, ca0.to_na)
+    assert_equal(na0, ca30.to_na)
+
+    #--- to_na!
+    assert_equal(na0, ca0.to_na!)
+    assert_equal(na0, ca30.to_na!)
+
+    #--- ca
+    assert_equal(ca0, na0.ca)
+
+    #--- to_ca
+    assert_equal(ca0, na0.to_ca)
+
+    #--- to_ca!
+    assert_equal(ca0, na0.to_ca!)
+    assert_equal(na0, ca30.to_na!)
+
+  end
+
+
+end

data/test/test_order.rb

@@ -0,0 +1,147 @@
+$:.unshift("../lib", "..")
+
+require 'test/unit'
+require 'carray'
+
+class TestCArrayOrder < Test::Unit::TestCase
+
+  def test_where
+    # ---
+    a = CArray.int(3,3).seq!
+    c = (a % 2).eq(0)
+    assert_equal(a[c], c.where)
+  end
+
+  def test_index
+    # ---
+    a = CArray.float(3,3).seq!
+    assert_equal(CA_INT32([[0,1,2],
+                           [3,4,5],
+                           [6,7,8]]), a.address)
+    assert_equal(CA_INT32([0,1,2]), a.index(0))
+    assert_equal(CA_INT32([0,1,2]), a.index(1))
+  end
+
+  def test_indices
+    # ---
+    a = CArray.float(3,3).seq!
+    a.indices {|x, y|
+      assert_equal(CA_INT32([[0,0,0],
+                             [1,1,1],
+                             [2,2,2]]), x)
+      assert_equal(CA_INT32([[0,1,2],
+                             [0,1,2],
+                             [0,1,2]]), y)
+    }
+  end
+
+  def test_reverse
+    a = CArray.int32(3,3).seq!
+    b = CArray.int32(3,3).seq!(8,-1)
+
+    # ---
+    assert_equal(b, a.reverse)
+    assert_equal(b, a.reversed)
+
+    c = a.to_ca.reverse!
+    assert_equal(b, c)
+  end
+
+  def test_sort
+    a = CA_INT32([[5,4,3],
+                  [0,1,2],
+                  [8,7,6]])
+    s = CA_INT32([[0,1,2],
+                  [3,4,5],
+                  [6,7,8]])
+    # ---
+    assert_equal(s, a.sort)
+    b = a.to_ca
+    b.sort!
+    assert_equal(s, b)
+  end
+
+  def test_sort_addr
+    a = CA_INT32([[5,4,3],
+                  [0,1,2],
+                  [8,7,6]])
+    # ---
+    assert_equal(CA_INT32([[3,4,5],
+                           [2,1,0],
+                           [8,7,6]]), a.sort_addr)
+    assert_equal(CA_INT32([[5,4,3],
+                           [0,1,2],
+                           [8,7,6]]), a.order)
+    # ---
+    assert_equal(CA_INT32([[6,7,8],
+                           [0,1,2],
+                           [5,4,3]]), a.sort_addr.reverse)
+    assert_equal(CA_INT32([[3,4,5],
+                           [8,7,6],
+                           [0,1,2]]), a.order(-1))
+  end
+
+  def test_uniq
+    a = CArray.int(100).random!(10)
+
+    # ---
+    assert_equal(a.to_a.uniq, a.uniq.to_a)
+
+  end
+
+  def test_search
+    # ---
+    a = CArray.int(9,9).seq!
+    assert_equal(50, a.bsearch(50))
+    assert_equal(50, a.search(50))
+    assert_equal([5,5], a.bsearch_index(50))
+    assert_equal([5,5], a.search_index(50))
+    # ---
+    a = CArray.int(9,9).seq!(0,2)
+    assert_equal(nil, a.bsearch(51))
+    assert_equal(nil, a.search(51))
+    assert_equal(nil, a.bsearch_index(51))
+    assert_equal(nil, a.search_index(51))
+    # ---
+    a = CArray.float(9,9).span!(0..80)
+    assert_equal(50, a.bsearch(50))
+    assert_equal(50, a.search(50))
+    assert_equal([5,5], a.bsearch_index(50))
+    assert_equal([5,5], a.search_index(50))
+
+    assert_equal(nil, a.bsearch(50.0001))
+    assert_equal(nil, a.search(50.0001))
+    assert_equal(nil, a.bsearch_index(50.0001))
+    assert_equal(nil, a.search_index(50.0001))
+    # ---
+    a = CArray.float(9,9).span!(0..80)    
+    assert_equal(50, a.search(50.0001, 0.0001))
+    assert_equal([5,5], a.search_index(50.0001, 0.0001))    
+
+    assert_equal(nil, a.search(50.0001, 0.00001))
+    assert_equal(nil, a.search_index(50.0001, 0.00001))   
+  end
+  
+  def test_search_nearest   
+    # ---
+    a = CArray.int(9,9).seq!
+    assert_equal(50, a.search_nearest(50))
+    assert_equal([5,5], a.search_nearest_index(50))
+    # ---
+    a = CArray.int(9,9).seq!(0,2)
+    assert_equal(25, a.search_nearest(51))
+    assert_equal([2,7], a.search_nearest_index(51))
+    # ---
+    a = CArray.float(9,9).span!(0..80)
+    assert_equal(50, a.search_nearest(50))
+    assert_equal([5,5], a.search_nearest_index(50))
+
+    assert_equal(50, a.search_nearest(50.5))
+    assert_equal([5,5], a.search_nearest_index(50.5))
+
+    # first detected
+    assert_equal(49, a.search_nearest(49.5))
+    assert_equal([5,4], a.search_nearest_index(49.5))
+    
+  end
+end

data/test/test_random.rb

@@ -0,0 +1,15 @@
+$:.unshift("../lib", "..")
+
+require 'test/unit'
+require 'carray'
+
+class TestCArrayRandom < Test::Unit::TestCase
+
+  def test_random
+    a = CArray.uint8(100, 100).random!(10)
+    assert_equal(0, (a < 0).count_true)
+    assert_equal(0, (a > 10).count_true)
+  end
+
+
+end

data/test/test_ref_store.rb

@@ -0,0 +1,211 @@
+$:.unshift("../lib", "..")
+
+require 'test/unit'
+require 'carray'
+
+class TestCArrayRefStore < Test::Unit::TestCase
+
+  def test_access_point
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(0, a[0,0])
+    assert_equal(4, a[1,1])
+    assert_equal(8, a[-1,-1])
+    assert_raise(IndexError) { a[0,3] }
+    assert_raise(IndexError) { a[-4,0] }
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[0,0] = -1
+    a[1,1] = -2
+    a[-1,-1] = -3
+    assert_equal(-1, a[0,0])
+    assert_equal(-2, a[1,1])
+    assert_equal(-3, a[-1,-1])
+    assert_raise(IndexError) { a[0,3] = -4 }
+    assert_raise(IndexError) { a[-4,0] = -5 }
+  end
+
+  def test_access_block
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(CA_INT([1, 4, 7]), a[0..-1,1])
+    assert_equal(CA_INT([[1], [4], [7]]), a[0..-1,[1]])
+    assert_equal(CABlock, a[0..2,0..2].class)
+    assert_equal(a, a[0..2,0..2])
+    assert_equal(CABlock, a[nil,nil].class)
+    assert_equal(a, a[nil,nil])
+    assert_equal(a, a[-3..-1,-3..-1])
+    assert_equal(CA_INT([[0,1,2],[3,4,5]]), a[0..1, nil])
+    assert_equal(CA_INT([[0,2],[3,5]]), a[0..1, [nil,2]])
+    assert_raise(IndexError) { a[0,0..3] }
+    assert_raise(IndexError) { a[-4..-1,0] }
+    assert_raise(IndexError) { a[0,[0,3,2]] }
+    assert_raise(IndexError) { a[0,[0,4]] }
+
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(CA_INT([[0,1,2], [3,4,5], [6,7,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[0..-1, 1] = 9
+    assert_equal(CA_INT([[0,9,2], [3,9,5], [6,9,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[0..-1, [1]] = 9
+    assert_equal(CA_INT([[0,9,2], [3,9,5], [6,9,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[nil, nil] = 9
+    assert_equal(CA_INT([[9,9,9], [9,9,9], [9,9,9]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[0..1, [nil,2]] = 9
+    assert_equal(CA_INT([[9,1,9], [9,4,9], [6,7,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_raise(IndexError) { a[0,0..3] = -3 }
+    assert_raise(IndexError) { a[-4..-1,0] = -4 }
+    assert_raise(IndexError) { a[0,[0,3,2]] = -5 }
+    assert_raise(IndexError) { a[0,[0,4]] = -6 }
+
+  end
+
+  def test_access_selection
+    # ---
+    a = CArray.int(3,3).seq!;
+    assert_equal(CA_INT([0,1,2,3,4]), a[a < 5])
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[a < 5] = 1
+    assert_equal(CA_INT([[1,1,1],[1,1,5],[6,7,8]]), a)
+  end
+
+  def test_access_mapping
+    # ---
+    a = CArray.int(3,3).seq!;
+    m = a.shuffle
+    assert_equal(10*m, (10*a)[m])
+
+    # ---
+    a = CArray.int(3,3).seq!
+    b = 10*a
+    m = a.shuffle
+    a[m] = (10*m)
+    assert_equal(b, a)
+  end
+
+  def test_access_grid
+    # ---
+    a = CArray.int(3,3).seq!;
+    assert_equal(CA_INT([[1,2],[4,5]]), a[CA_INT([0,1]),1..2])
+    assert_equal(CA_INT([[0,2],[6,8]]), a[+[0,2],+[0,2]])
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[+[0,2],+[0,2]] = 9
+    assert_equal(CA_INT([[9,1,9], [3,4,5], [9,7,9]]), a)
+  end
+
+  def test_access_repeat
+    # ---
+    # See test_CARepeat
+  end
+
+  def test_access_method
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(CA_INT([0,1,2,3,4]), a[:lt, 5])
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[:lt,5] = 1
+    assert_equal(CA_INT([[1,1,1],[1,1,5],[6,7,8]]), a)
+  end
+
+  def test_access_address
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(0, a[0])
+    assert_equal(4, a[4])
+    assert_equal(8, a[-1])
+    assert_raise(IndexError) { a[9] }
+    assert_raise(IndexError) { a[-10] }
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[0] = -1
+    a[4] = -2
+    a[-1] = -3
+    assert_equal(-1, a[0,0])
+    assert_equal(-2, a[1,1])
+    assert_equal(-3, a[-1,-1])
+    assert_raise(IndexError) { a[9] = -4 }
+    assert_raise(IndexError) { a[-10] = -5 }
+  end
+
+  def test_access_address_block
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(CA_INT([4]), a[[4]])
+    assert_equal(CA_INT([0,1,2,3]), a[[0..3]])
+    assert_equal(CABlock, a[nil].class)
+    assert_equal(CA_INT([0,1,2,3,4,5,6,7,8]), a[nil])
+    assert_raise(IndexError) { a[0..9] }
+    assert_raise(IndexError) { a[-10..-1] }
+
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_equal(CA_INT([[0,1,2], [3,4,5], [6,7,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[[1..-1,3]] = 9
+    assert_equal(CA_INT([[0,9,2], [3,9,5], [6,9,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[[1,3,3]] = 9
+    assert_equal(CA_INT([[0,9,2], [3,9,5], [6,9,8]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[nil] = 9
+    assert_equal(CA_INT([[9,9,9], [9,9,9], [9,9,9]]), a)
+
+    # ---
+    a = CArray.int(3,3).seq!
+    assert_raise(IndexError) { a[0..9] = -3 }
+    assert_raise(IndexError) { a[-10..-1] = -4 }
+  end
+
+  def test_access_address_selection
+    # ---
+    a = CArray.int(3,3).seq!;
+    a1 = a.flatten
+    assert_equal(CA_INT([0,1,2,3,4]), a[a1 < 5])
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[a1 < 5] = 1
+    assert_equal(CA_INT([[1,1,1],[1,1,5],[6,7,8]]), a)
+  end
+
+  def test_access_address_grid
+    # ---
+    a = CArray.int(3,3).seq!;
+    assert_equal(CA_INT([0,2,4,6,8]), a[+[0,2,4,6,8]])
+
+    # ---
+    a = CArray.int(3,3).seq!
+    a[+[0,2,4,6,8]] = 9
+    assert_equal(CA_INT([[9,1,9],[3,9,5],[9,7,9]]), a)
+  end
+
+end