class TestImageCollection(TestCase): pattern = [ os.path.join(data_dir, pic) for pic in ['camera.png', 'color.png'] ] pattern_matched = [ os.path.join(data_dir, pic) for pic in ['camera.png', 'moon.png'] ] def setUp(self): reset_plugins() # Generic image collection with images of different shapes. self.images = ImageCollection(self.pattern) # Image collection with images having shapes that match. self.images_matched = ImageCollection(self.pattern_matched) def test_len(self): assert len(self.images) == 2 def test_getitem(self): num = len(self.images) for i in range(-num, num): assert isinstance(self.images[i], np.ndarray) assert_allclose(self.images[0], self.images[-num]) def return_img(n): return self.images[n] with testing.raises(IndexError): return_img(num) with testing.raises(IndexError): return_img(-num - 1) def test_slicing(self): assert type(self.images[:]) is ImageCollection assert len(self.images[:]) == 2 assert len(self.images[:1]) == 1 assert len(self.images[1:]) == 1 assert_allclose(self.images[0], self.images[:1][0]) assert_allclose(self.images[1], self.images[1:][0]) assert_allclose(self.images[1], self.images[::-1][0]) assert_allclose(self.images[0], self.images[::-1][1]) def test_files_property(self): assert isinstance(self.images.files, list) def set_files(f): self.images.files = f with testing.raises(AttributeError): set_files('newfiles') def test_custom_load(self): load_pattern = [(1, 'one'), (2, 'two')] def load_fn(x): return x ic = ImageCollection(load_pattern, load_func=load_fn) assert_equal(ic[1], (2, 'two')) def test_custom_load_func(self): def load_fn(x): return x ic = ImageCollection(os.pathsep.join(self.pattern), load_func=load_fn) assert_equal(ic[0], self.pattern[0]) def test_concatenate(self): array = self.images_matched.concatenate() expected_shape = (len(self.images_matched), ) + self.images[0].shape assert_equal(array.shape, expected_shape) def test_concatentate_mismatched_image_shapes(self): with testing.raises(ValueError): self.images.concatenate()
class TestImageCollection(): pattern = [os.path.join(data_dir, pic) for pic in ['camera.png', 'color.png']] pattern_matched = [os.path.join(data_dir, pic) for pic in ['camera.png', 'moon.png']] def setUp(self): reset_plugins() # Generic image collection with images of different shapes. self.images = ImageCollection(self.pattern) # Image collection with images having shapes that match. self.images_matched = ImageCollection(self.pattern_matched) def test_len(self): assert len(self.images) == 2 def test_getitem(self): num = len(self.images) for i in range(-num, num): assert type(self.images[i]) is np.ndarray assert_allclose(self.images[0], self.images[-num]) # assert_raises expects a callable, hence this thin wrapper function. def return_img(n): return self.images[n] assert_raises(IndexError, return_img, num) assert_raises(IndexError, return_img, -num - 1) def test_slicing(self): assert type(self.images[:]) is ImageCollection assert len(self.images[:]) == 2 assert len(self.images[:1]) == 1 assert len(self.images[1:]) == 1 assert_allclose(self.images[0], self.images[:1][0]) assert_allclose(self.images[1], self.images[1:][0]) assert_allclose(self.images[1], self.images[::-1][0]) assert_allclose(self.images[0], self.images[::-1][1]) def test_files_property(self): assert isinstance(self.images.files, list) def set_files(f): self.images.files = f assert_raises(AttributeError, set_files, 'newfiles') def test_custom_load(self): load_pattern = [(1, 'one'), (2, 'two')] def load_fn(x): return x ic = ImageCollection(load_pattern, load_func=load_fn) assert_equal(ic[1], (2, 'two')) def test_custom_load_func(self): def load_fn(x): return x ic = ImageCollection(os.pathsep.join(self.pattern), load_func=load_fn) assert_equal(ic[0], self.pattern[0]) def test_concatenate(self): array = self.images_matched.concatenate() expected_shape = (len(self.images_matched),) + self.images[0].shape assert_equal(array.shape, expected_shape) def test_concatentate_mismatched_image_shapes(self): assert_raises(ValueError, self.images.concatenate)
class TestImageCollection(TestCase): pattern = [ os.path.join(data_dir, pic) for pic in ['brick.png', 'color.png'] ] pattern_matched = [ os.path.join(data_dir, pic) for pic in ['brick.png', 'moon.png'] ] def setUp(self): reset_plugins() # Generic image collection with images of different shapes. self.images = ImageCollection(self.pattern) # Image collection with images having shapes that match. self.images_matched = ImageCollection(self.pattern_matched) # Same images as a collection of frames self.frames_matched = MultiImage(self.pattern_matched) def test_len(self): assert len(self.images) == 2 def test_getitem(self): num = len(self.images) for i in range(-num, num): assert isinstance(self.images[i], np.ndarray) assert_allclose(self.images[0], self.images[-num]) def return_img(n): return self.images[n] with testing.raises(IndexError): return_img(num) with testing.raises(IndexError): return_img(-num - 1) def test_slicing(self): assert type(self.images[:]) is ImageCollection assert len(self.images[:]) == 2 assert len(self.images[:1]) == 1 assert len(self.images[1:]) == 1 assert_allclose(self.images[0], self.images[:1][0]) assert_allclose(self.images[1], self.images[1:][0]) assert_allclose(self.images[1], self.images[::-1][0]) assert_allclose(self.images[0], self.images[::-1][1]) def test_files_property(self): assert isinstance(self.images.files, list) def set_files(f): self.images.files = f with testing.raises(AttributeError): set_files('newfiles') def test_custom_load_func_w_kwarg(self): load_pattern = os.path.join(data_dir, 'no_time_for_that_tiny.gif') def load_fn(f, step): vid = imageio.get_reader(f) seq = [v for v in vid.iter_data()] return seq[::step] ic = ImageCollection(load_pattern, load_func=load_fn, step=3) # Each file should map to one image (array). assert len(ic) == 1 # GIF file has 24 frames, so 24 / 3 equals 8. assert len(ic[0]) == 8 def test_custom_load_func(self): def load_fn(x): return x ic = ImageCollection(os.pathsep.join(self.pattern), load_func=load_fn) assert_equal(ic[0], self.pattern[0]) def test_concatenate(self): array = self.images_matched.concatenate() expected_shape = (len(self.images_matched), ) + self.images[0].shape assert_equal(array.shape, expected_shape) def test_concatenate_mismatched_image_shapes(self): with testing.raises(ValueError): self.images.concatenate() def test_multiimage_imagecollection(self): assert_equal(self.images_matched[0], self.frames_matched[0]) assert_equal(self.images_matched[1], self.frames_matched[1])