Add test

cupy · Aug 9, 2017 · 8305116 · 8305116
1 parent 5a76371
commit 8305116
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/cupy_tests/sparse_tests/test_dia.py b/tests/cupy_tests/sparse_tests/test_dia.py
@@ -52,6 +52,23 @@ def test_ndim(self):
     def test_nnz(self):
         self.assertEqual(self.m.nnz, 5)
 
+    @unittest.skipUnless(scipy_available, 'requires scipy')
+    def test_str(self):
+        self.assertEqual(str(self.m), '''  (1, 1)\t1.0
+  (2, 2)\t2.0
+  (1, 0)\t3.0
+  (2, 1)\t4.0''')
+
+    def test_toarray(self):
+        m = self.m.toarray()
+        expect = [
+            [0, 0, 0, 0],
+            [3, 1, 0, 0],
+            [0, 4, 2, 0]
+        ]
+        self.assertTrue(m.flags.c_contiguous)
+        cupy.testing.assert_allclose(m, expect)
+
 
 @testing.parameterize(*testing.product({
     'dtype': [numpy.float32, numpy.float64],