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())
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)))
def test_get_dtype_name_float(self): self.assertEqual(cb.get_dtype_name("float32"), "float32") self.assertEqual(cb.get_dtype_name(np.float32), "float32")
def test_get_default_dtype(self): self.assertEqual(cb.get_dtype_name(cb.get_default_dtype()), "float32")
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)))