mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 19:17:18 +08:00
Steven Silvester
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
import os
|
|
|
|
import numpy as np
|
|
from numpy.testing import assert_raises, assert_equal, assert_allclose
|
|
|
|
from skimage import data_dir
|
|
from skimage.io.collection import MultiImage, ImageCollection
|
|
|
|
import six
|
|
|
|
|
|
class TestMultiImage():
|
|
|
|
def setUp(self):
|
|
# This multipage TIF file was created with imagemagick:
|
|
# convert im1.tif im2.tif -adjoin multipage.tif
|
|
paths = [os.path.join(data_dir, 'multipage.tif'),
|
|
os.path.join(data_dir, 'no_time_for_that.gif')]
|
|
self.imgs = [MultiImage(paths[0]),
|
|
MultiImage(paths[0], conserve_memory=False),
|
|
MultiImage(paths[1]),
|
|
MultiImage(paths[1], conserve_memory=False),
|
|
ImageCollection(paths[0]),
|
|
ImageCollection(paths[1], conserve_memory=False),
|
|
ImageCollection('%s:%s' % (paths[0], paths[1]))]
|
|
|
|
def test_len(self):
|
|
assert len(self.imgs[0]) == len(self.imgs[1]) == 2
|
|
assert len(self.imgs[2]) == len(self.imgs[3]) == 24
|
|
assert len(self.imgs[4]) == 2
|
|
assert len(self.imgs[5]) == 24
|
|
assert len(self.imgs[6]) == 26
|
|
|
|
def test_slicing(self):
|
|
img = self.imgs[-1]
|
|
assert type(img[:]) is ImageCollection
|
|
assert len(img[:]) == 26
|
|
assert len(img[:1]) == 1
|
|
assert len(img[1:]) == 25
|
|
assert_allclose(img[0], img[:1][0])
|
|
assert_allclose(img[1], img[1:][0])
|
|
assert_allclose(img[-1], img[::-1][0])
|
|
assert_allclose(img[0], img[::-1][-1])
|
|
|
|
def test_getitem(self):
|
|
for img in self.imgs:
|
|
num = len(img)
|
|
|
|
for i in range(-num, num):
|
|
assert type(img[i]) is np.ndarray
|
|
assert_allclose(img[0], img[-num])
|
|
|
|
# assert_raises expects a callable, hence this thin wrapper function.
|
|
def return_img(n):
|
|
return img[n]
|
|
assert_raises(IndexError, return_img, num)
|
|
assert_raises(IndexError, return_img, -num - 1)
|
|
|
|
def test_files_property(self):
|
|
for img in self.imgs:
|
|
if isinstance(img, ImageCollection):
|
|
continue
|
|
|
|
assert isinstance(img.filename, six.string_types)
|
|
|
|
def set_filename(f):
|
|
img.filename = f
|
|
assert_raises(AttributeError, set_filename, 'newfile')
|
|
|
|
def test_conserve_memory_property(self):
|
|
for img in self.imgs:
|
|
assert isinstance(img.conserve_memory, bool)
|
|
|
|
def set_mem(val):
|
|
img.conserve_memory = val
|
|
assert_raises(AttributeError, set_mem, True)
|
|
|
|
def test_concatenate(self):
|
|
for img in self.imgs:
|
|
array = img.concatenate()
|
|
assert_equal(array.shape, (len(img),) + img[0].shape)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from numpy.testing import run_module_suite
|
|
run_module_suite()
|