def test_pickleable(self):
obj = Image((3, 3, 3))
round_trip = pickle.loads(pickle.dumps(obj))
assert round_trip
assert round_trip.shape == obj.shape
assert np.array_equal(round_trip.bounds[0], obj.bounds[0])
assert np.array_equal(round_trip.bounds[1], obj.bounds[1])
def test_dtype(self):
img = Image((1, 1, 1))
assert img.dtype == np.uint8
def test_invalid_shape(self):
with self.assertRaises(AssertionError):
Image((1, 1, 1, 1))
with self.assertRaises(TypeError):
Image()
def test_concat(self):
img1 = Image((5, 5, 3))
img2 = Image((10, 10, 3))
concat_image = img1.concat(img2)
assert concat_image.flat_dim == img1.flat_dim + img2.flat_dim