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))
