def test_iter(): imgiter = iter(gimg) for data in imgiter: yield assert_equal, data.shape, (3,2) tmp = Image(np.zeros(gimg.shape), gimg.coordmap) write_data(tmp, data_generator(gimg, range(gimg.shape[0]))) yield assert_true, np.allclose(np.asarray(tmp), np.asarray(gimg)) tmp = Image(np.zeros(gimg.shape), gimg.coordmap) g = data_generator(gimg) write_data(tmp, g) yield assert_true, np.allclose(np.asarray(tmp), np.asarray(gimg))
def test_write_slices(self): tmp = np.zeros(self.img.shape) write_data(tmp, slice_generator(self.img)) assert_almost_equal(tmp, np.asarray(self.img)) tmp = np.zeros(self.img.shape) write_data(tmp, slice_generator(self.img, axis=1)) assert_almost_equal(tmp, np.asarray(self.img)) tmp = np.zeros(self.img.shape) write_data(tmp, slice_generator(self.img, axis=2)) assert_almost_equal(tmp, np.asarray(self.img))
def test_multi_slice_write(self): a = np.zeros(self.img.shape) write_data(a, slice_generator(self.img, axis=[0, 1])) assert_almost_equal(a, np.asarray(self.img))