mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-08 11:58:33 +08:00
Juan Nunez-Iglesias
a61ac37ff2
Previously we added the ImageCollection.concatenate function. This updates MultiImage to have the same functionality, and moves the grunt work to an outside function to avoid repetition.
123 lines
3.7 KiB
Python
123 lines
3.7 KiB
Python
import sys
|
|
import os.path
|
|
|
|
import numpy as np
|
|
from numpy.testing import *
|
|
from numpy.testing.decorators import skipif
|
|
|
|
from skimage import data_dir
|
|
from skimage.io import ImageCollection, MultiImage
|
|
|
|
|
|
try:
|
|
from PIL import Image
|
|
except ImportError:
|
|
PIL_available = False
|
|
else:
|
|
PIL_available = True
|
|
|
|
if sys.version_info[0] > 2:
|
|
basestring = str
|
|
|
|
|
|
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):
|
|
self.collection = ImageCollection(self.pattern)
|
|
self.collection_matched = ImageCollection(self.pattern_matched)
|
|
|
|
def test_len(self):
|
|
assert len(self.collection) == 2
|
|
|
|
def test_getitem(self):
|
|
num = len(self.collection)
|
|
for i in range(-num, num):
|
|
assert type(self.collection[i]) is np.ndarray
|
|
assert_array_almost_equal(self.collection[0],
|
|
self.collection[-num])
|
|
|
|
#assert_raises expects a callable, hence this do-very-little func
|
|
def return_img(n):
|
|
return self.collection[n]
|
|
assert_raises(IndexError, return_img, num)
|
|
assert_raises(IndexError, return_img, -num - 1)
|
|
|
|
def test_files_property(self):
|
|
assert isinstance(self.collection.files, list)
|
|
|
|
def set_files(f):
|
|
self.collection.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_concatenate(self):
|
|
ar = self.collection_matched.concatenate()
|
|
assert_equal(ar.shape, (len(self.collection_matched),) +
|
|
self.collection[0].shape)
|
|
assert_raises(ValueError, self.collection.concatenate)
|
|
|
|
|
|
class TestMultiImage():
|
|
|
|
def setUp(self):
|
|
# This multipage TIF file was created with imagemagick:
|
|
# convert im1.tif im2.tif -adjoin multipage.tif
|
|
if PIL_available:
|
|
self.img = MultiImage(os.path.join(data_dir, 'multipage.tif'))
|
|
|
|
@skipif(not PIL_available)
|
|
def test_len(self):
|
|
assert len(self.img) == 2
|
|
|
|
@skipif(not PIL_available)
|
|
def test_getitem(self):
|
|
num = len(self.img)
|
|
for i in range(-num, num):
|
|
assert type(self.img[i]) is np.ndarray
|
|
assert_array_almost_equal(self.img[0],
|
|
self.img[-num])
|
|
|
|
#assert_raises expects a callable, hence this do-very-little func
|
|
def return_img(n):
|
|
return self.img[n]
|
|
assert_raises(IndexError, return_img, num)
|
|
assert_raises(IndexError, return_img, -num - 1)
|
|
|
|
@skipif(not PIL_available)
|
|
def test_files_property(self):
|
|
assert isinstance(self.img.filename, basestring)
|
|
|
|
def set_filename(f):
|
|
self.img.filename = f
|
|
assert_raises(AttributeError, set_filename, 'newfile')
|
|
|
|
@skipif(not PIL_available)
|
|
def test_conserve_memory_property(self):
|
|
assert isinstance(self.img.conserve_memory, bool)
|
|
|
|
def set_mem(val):
|
|
self.img.conserve_memory = val
|
|
assert_raises(AttributeError, set_mem, True)
|
|
|
|
@skipif(not PIL_available)
|
|
def test_concatenate(self):
|
|
ar = self.img.concatenate()
|
|
assert_equal(ar.shape, (len(self.img),) +
|
|
self.img[0].shape)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_module_suite()
|