def setUp(self): self.shape = Shape("3x4x5") self.l0 = cb.linear_cube(self.shape, start=0.5, increment=1.5) self.l1 = cb.linear_cube(self.shape, start=0.5, increment=0.5) self.l2 = cb.linear_cube(self.shape, start=-1.0, increment=0.5) self.epsilon = 1.0e-8 e = self.epsilon self.c0 = np.array([0.0, 0.0, e, 1.0, 1.0 + e, 1.0], ) self.c1 = np.array([0.0, e, 0.0, 1.0, 1.0, 1.0 + e])
def setUp(self): self.shape = Shape("3x4x5") self.l0 = cb.linear_cube(self.shape, start=0.5, increment=1.5) self.l1 = cb.linear_cube(self.shape, start=0.5, increment=0.5) self.l2 = cb.linear_cube(self.shape, start=-1.0, increment=0.5) self.epsilon = 1.0e-8 e = self.epsilon self.c0 = np.array([0.0, 0.0, e, 1.0, 1.0 + e, 1.0]) self.c1 = np.array([0.0, e, 0.0, 1.0, 1.0, 1.0 + e])
def join3(self): cube0 = cb.const_cube(shape="3x5", value=1.5) cube1 = cb.linear_cube(shape="3x5", start=1.5, increment=0.5) cube2 = cb.const_cube(shape="3x5", value=-1.5) cubes = (cube0, cube1, cube2) result = np.array(cubes) return self.impl_join(cubes, result)
def create_linear_cubes(self, shape, dtype): cube_l = cb.linear_cube(shape=shape, dtype=dtype) cube_r = cube_l * 1.5 dmin = 2 for d in shape: assert d >= dmin, "d={} < {}".format(d, dmin) index_diff_0 = tuple(0 for i in shape) index_diff_1 = tuple(1 for i in shape) cube_r[index_diff_0] = cube_l[index_diff_0] * 0.1 cube_r[index_diff_1] = cube_l[index_diff_1] * 10.3 return cube_l, cube_r
def impl_read_write(self, shape, dtype, file_format): dtype = cb.get_dtype(dtype) shape = Shape(shape) out0_filename_format = "o0tmp_{shape}_{dtype}.{format}" out0_filename = out0_filename_format.format(shape=shape, dtype=dtype.__name__, format='raw') out1_filename_format = "o1tmp_{shape}_{dtype}.{format}" out1_filename = out1_filename_format.format(shape=shape, dtype=dtype.__name__, format=file_format) out2_filename_format = "o2tmp_{shape}_{dtype}.{format}" out2_filename = out2_filename_format.format(shape=shape, dtype=dtype.__name__, format='raw') cube = cb.linear_cube(shape=shape, dtype=dtype) cb.write_cube_raw(cube=cube, file=out0_filename_format) self.assertFileExistsAndHasShape(out0_filename, shape, dtype=dtype) returncode, output, error = self.run_program( """-i '{i}' -It '{t}' -s '{s}' -e 'write_cube(cube=_r, filename="{o}", format="{f}", dtype="{t}")'""" .format(i=out0_filename_format, s=shape, o=out1_filename_format, f=file_format, t=dtype.__name__)) self.assertFileExists(out1_filename) returncode, output, error = self.run_program( """-e 'read_cube(shape="{s}", filename="{i}", format="{f}", dtype="{t}")' -e 'write_cube(cube=_r, filename="{o}", format="raw", dtype="{t}")'""" .format(i=out1_filename_format, s=shape, o=out2_filename_format, f=file_format, t=dtype.__name__)) self.assertFileExistsAndHasShape(out2_filename, shape, dtype=dtype) self.assertFilesAreEqual(out0_filename, out2_filename) self.remove_files(out0_filename, out1_filename, out2_filename)
def impl_read_write(self, shape, dtype, file_format): dtype = cb.get_dtype(dtype) shape = Shape(shape) out0_filename_format = "o0tmp_{shape}_{dtype}.{format}" out0_filename = out0_filename_format.format(shape=shape, dtype=dtype.__name__, format='raw') out1_filename_format = "o1tmp_{shape}_{dtype}.{format}" out1_filename = out1_filename_format.format(shape=shape, dtype=dtype.__name__, format=file_format) out2_filename_format = "o2tmp_{shape}_{dtype}.{format}" out2_filename = out2_filename_format.format(shape=shape, dtype=dtype.__name__, format='raw') cube = cb.linear_cube(shape=shape, dtype=dtype) cb.write_cube_raw(cube=cube, file=out0_filename_format) self.assertFileExistsAndHasShape(out0_filename, shape, dtype=dtype) returncode, output, error = self.run_program( """-i '{i}' -It '{t}' -s '{s}' -e 'write_cube(cube=_r, filename="{o}", format="{f}", dtype="{t}")'""".format( i=out0_filename_format, s=shape, o=out1_filename_format, f=file_format, t=dtype.__name__)) self.assertFileExists(out1_filename) returncode, output, error = self.run_program( """-e 'read_cube(shape="{s}", filename="{i}", format="{f}", dtype="{t}")' -e 'write_cube(cube=_r, filename="{o}", format="raw", dtype="{t}")'""".format( i=out1_filename_format, s=shape, o=out2_filename_format, f=file_format, t=dtype.__name__)) self.assertFileExistsAndHasShape(out2_filename, shape, dtype=dtype) self.assertFilesAreEqual(out0_filename, out2_filename) self.remove_files(out0_filename, out1_filename, out2_filename)