def setUp(self): self.data = np.empty(shape=(128, 256, 4), dtype='uint8') self.data[:, :, 0] = np.arange(256) self.data[:, :, 1] = np.arange(128)[:, np.newaxis] self.data[:, :, 2] = np.arange(256)[::-1] self.data[:, :, 3] = np.arange(128)[::-1, np.newaxis] self.image_24 = Image(self.data[..., :3]) self.image_32 = Image(self.data)
def test_bounds_overrride(self): image = Image(self.data, bounds=[200, 100]) self.assertEqual(image.bounds, [200, 100])
def test_init_bad_dtype(self): data = np.array(['red'] * 65536).reshape(128, 128, 4) with self.assertRaises(TraitError): Image(data=data)
def test_init_bad_shape(self): data = np.zeros(shape=(256, 256), dtype='uint8') with self.assertRaises(TraitError): Image(data=data)