cumo 0.5.0 → 0.5.2

data/test/narray_test.rb

@@ -736,10 +736,90 @@ class NArrayTest < Test::Unit::TestCase
       at.at([0, 1], [0, 1]).inplace - 1
       assert { at == [[0, 2, 3], [4, 4, 6]] }
     end
+
+    sub_test_case "#{dtype}.from_binary" do
+      test "frozen string" do
+        shape = [2, 5]
+        a = dtype.new(*shape)
+        a.rand(0, 10)
+        original_data = a.to_binary
+        data = original_data.dup.freeze
+        restored_a = dtype.from_binary(data, shape)
+        assert { restored_a == a }
+        restored_a[0, 0] += 1
+        assert { restored_a != a }
+        assert { data == original_data }
+      end
+
+      test "not frozen string" do
+        shape = [2, 5]
+        a = dtype.new(*shape)
+        a.rand(0, 10)
+        original_data = a.to_binary
+        data = original_data.dup
+        restored_a = dtype.from_binary(data, shape)
+        assert { restored_a == a }
+        restored_a[0, 0] += 1
+        assert { restored_a != a }
+        assert { data == original_data }
+      end
+    end
+
+    sub_test_case "#{dtype}#store_binary" do
+      test "frozen string" do
+        shape = [2, 5]
+        a = dtype.new(*shape)
+        a.rand(0, 10)
+        original_data = a.to_binary
+        data = original_data.dup.freeze
+        restored_a = dtype.new(*shape)
+        restored_a.store_binary(data)
+        assert { restored_a == a }
+        restored_a[0, 0] += 1
+        assert { restored_a != a }
+        assert { data == original_data }
+      end
+
+      test "not frozen string" do
+        shape = [2, 5]
+        a = dtype.new(*shape)
+        a.rand(0, 10)
+        original_data = a.to_binary
+        data = original_data.dup
+        restored_a = dtype.new(*shape)
+        restored_a.store_binary(data)
+        assert { restored_a == a }
+        restored_a[0, 0] += 1
+        assert { restored_a != a }
+        assert { data == original_data }
+      end
+    end
   end
 
   test "Cumo::DFloat.cast(Cumo::RObject[1, nil, 3])" do
     assert_equal(Cumo::DFloat[1, Float::NAN, 3].format_to_a,
                  Cumo::DFloat.cast(Cumo::RObject[1, nil, 3]).format_to_a)
   end
+
+  test "single element array" do
+    assert { Cumo::SFloat[1].mean == 1.0 }
+    assert { Cumo::DFloat[1].mean == 1.0 }
+    assert { Cumo::SComplex[1].mean == 1.0 }
+    assert { Cumo::DComplex[1].mean == 1.0 }
+
+    assert { Cumo::SFloat[1].var.to_f.nan? }
+    assert { Cumo::DFloat[1].var.to_f.nan? }
+    assert { Cumo::SComplex[1].var.to_f.nan? }
+    assert { Cumo::DComplex[1].var.to_f.nan? }
+
+    assert { Cumo::SFloat[1].stddev.to_f.nan? }
+    assert { Cumo::DFloat[1].stddev.to_f.nan? }
+    assert { Cumo::SComplex[1].stddev.to_f.nan? }
+    assert { Cumo::DComplex[1].stddev.to_f.nan? }
+
+    assert { Cumo::SFloat[1].rms == 1.0 }
+    assert { Cumo::DFloat[1].rms == 1.0 }
+    assert { Cumo::SComplex[1].rms == 1.0 }
+    assert { Cumo::DComplex[1].rms == 1.0 }
+  end
 end

data/test/ractor_test.rb

@@ -10,7 +10,7 @@ class NArrayRactorTest < CumoTestBase
       dtype = data.fetch(:dtype)
       ary = random_array(dtype)
       r = Ractor.new(ary) { |x| x }
-      ary2 = r.take
+      ary2 = r.respond_to?(:take) ? r.take : r.value
       assert_equal(ary, ary2)
       assert_not_same(ary, ary2)
     end
@@ -22,7 +22,7 @@ class NArrayRactorTest < CumoTestBase
       r = Ractor.new(ary1) do |ary2|
         [ary2, ary2 * 10]
       end
-      ary2, res = r.take
+      ary2, res = r.respond_to?(:take) ? r.take : r.value
       assert_equal((dtype != Cumo::RObject),
                    ary1.equal?(ary2))
       assert_equal(ary1 * 10, res)
@@ -37,7 +37,9 @@ class NArrayRactorTest < CumoTestBase
       r2 = Ractor.new(ary1) do |ary4|
         ary4 * 10
       end
-      assert_equal(r1.take, r2.take)
+      result1 = r1.respond_to?(:take) ? r1.take : r1.value
+      result2 = r2.respond_to?(:take) ? r2.take : r2.value
+      assert_equal(result1, result2)
     end
 
     def random_array(dtype, n=1000)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cumo
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.5.2
 platform: ruby
 authors:
 - Naotoshi Seo
@@ -35,6 +35,7 @@ files:
 - 3rd_party/mkmf-cu/test/test_mkmf-cu.rb
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
+- Dockerfile
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -46,6 +47,8 @@ files:
 - bin/console
 - bin/setup
 - cumo.gemspec
+- docker-build.sh
+- docker-launch.sh
 - docs/src-tree.md
 - ext/cumo/cuda/cublas.c
 - ext/cumo/cuda/cudnn.c
@@ -328,7 +331,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.2
+rubygems_version: 4.0.4
 specification_version: 4
 summary: Cumo is CUDA aware numerical library whose interface is highly compatible
   with Ruby Numo