def test__setattr__(self): """ Tests :attr:`colour_hdri.utilities.image.ImageStack.__getattr__` method. """ image_stack = ImageStack().from_files(self._test_jpg_images) self.assertTupleEqual(image_stack.data.shape, (426, 640, 3, 3)) image_stack.data = np.random.random((20, 10, 3, 3)) self.assertTupleEqual(image_stack.data.shape, (20, 10, 3, 3)) np.testing.assert_almost_equal( image_stack.f_number, np.array([8, 8, 8]), decimal=7) image_stack.f_number = np.array([1, 2, 3]) np.testing.assert_almost_equal( image_stack.f_number, np.array([1, 2, 3]), decimal=7) self.assertEqual(image_stack[0].metadata.f_number, 1) self.assertListEqual(list(image_stack.black_level), [None, None, None]) image_stack.black_level = np.array([2048, 2048, 2048]) np.testing.assert_almost_equal( image_stack.black_level, np.array([2048, 2048, 2048]), decimal=7) self.assertEqual(image_stack[0].metadata.black_level, 2048)
def test_from_files(self): """ Tests :attr:`colour_hdri.utilities.image.ImageStack.test_from_files` method. """ image_stack = ImageStack().from_files(reversed(self._test_jpg_images)) self.assertListEqual(list(image_stack.path), self._test_jpg_images)
def setUp(self): """ Initialises common tests attributes. """ self._test_jpg_images = filter_files(FROBISHER_001_DIRECTORY, ('jpg', )) self._image_stack = ImageStack().from_files(self._test_jpg_images)
def test__delitem__(self): """ Tests :attr:`colour_hdri.utilities.image.ImageStack.__delitem__` method. """ image_stack = ImageStack().from_files(self._test_jpg_images) del image_stack[0] self.assertEqual(len(image_stack), 2)
def test__setitem__(self): """ Tests :attr:`colour_hdri.utilities.image.ImageStack.__setitem__` method. """ image_stack = ImageStack() image = Image(self._test_jpg_images[0]) image.read_data() image.read_metadata() image_stack.insert(0, image) self.assertEqual(image_stack[0], image)