Example #1
0
 def read_cube(self):
     a = np.array([[1.0, -1.3], [1.3, -0.2]], dtype=cb.get_default_dtype())
     a.tofile("file_a.raw")
     returncode, output, error = self.run_program("""-e 'read_cube(filename="file_a.raw", shape=("{s}"), dtype="{t!r}")' '_r.sum()' --print""".format(
         s=Shape(a.shape),
         t=cb.get_dtype_name(a.dtype),
     ))
     self.assertEqual(returncode, 0)
     v = float(output.strip())
     self.assertAlmostEqual(v, a.sum())
Example #2
0
 def test_get_dtype_name_int(self):
     a = np.array([1.5, 2.4], dtype=np.float64)
     self.assertEqual(
         cb.get_dtype_name(cb.as_dtype(a, dtype="int32").dtype), "int32")
     self.assertEqual(cb.get_dtype_name(cb.as_dtype(a).dtype),
                      cb.get_dtype_name(cb.get_dtype(None)))
Example #3
0
 def test_get_dtype_name_float(self):
     self.assertEqual(cb.get_dtype_name("float32"), "float32")
     self.assertEqual(cb.get_dtype_name(np.float32), "float32")
Example #4
0
 def test_get_default_dtype(self):
     self.assertEqual(cb.get_dtype_name(cb.get_default_dtype()), "float32")
Example #5
0
 def test_get_dtype_name_int(self):
     a = np.array([1.5, 2.4], dtype=np.float64)
     self.assertEqual(cb.get_dtype_name(cb.as_dtype(a, dtype="int32").dtype), "int32")
     self.assertEqual(cb.get_dtype_name(cb.as_dtype(a).dtype), cb.get_dtype_name(cb.get_dtype(None)))
Example #6
0
 def test_get_dtype_name_float(self):
     self.assertEqual(cb.get_dtype_name("float32"), "float32")
     self.assertEqual(cb.get_dtype_name(np.float32), "float32")
Example #7
0
 def test_get_default_dtype(self):
     self.assertEqual(cb.get_dtype_name(cb.get_default_dtype()), "float32")