wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-08 16:32:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add a multipage tiff test and cleanup test_tifffile

Browse Source
This commit is contained in:
Steven Silvester
2015-05-12 21:10:23 -05:00
parent e240dff395
commit dfaab1957e
1 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/io/tests/test_tifffile.py
+17 -13
View File
@@ -1,6 +1,6 @@
import os
import skimage as si
import skimage.io as sio
from ... import data_dir
from .. import imread, imsave, use_plugin, reset_plugins
import numpy as np
from numpy.testing import (
@@ -8,38 +8,42 @@ from numpy.testing import (
from tempfile import NamedTemporaryFile
_plugins = sio.plugin_order()
sio.use_plugin('tifffile')
np.random.seed(0)
def setup():
use_plugin('tifffile')
np.random.seed(0)
def teardown():
sio.reset_plugins()
reset_plugins()
def test_imread_uint16():
expected = np.load(os.path.join(si.data_dir, 'chessboard_GRAY_U8.npy'))
img = sio.imread(os.path.join(si.data_dir, 'chessboard_GRAY_U16.tif'))
expected = np.load(os.path.join(data_dir, 'chessboard_GRAY_U8.npy'))
img = imread(os.path.join(data_dir, 'chessboard_GRAY_U16.tif'))
assert img.dtype == np.uint16
assert_array_almost_equal(img, expected)
def test_imread_uint16_big_endian():
expected = np.load(os.path.join(si.data_dir, 'chessboard_GRAY_U8.npy'))
img = sio.imread(os.path.join(si.data_dir, 'chessboard_GRAY_U16B.tif'))
expected = np.load(os.path.join(data_dir, 'chessboard_GRAY_U8.npy'))
img = imread(os.path.join(data_dir, 'chessboard_GRAY_U16B.tif'))
assert img.dtype == np.uint16
assert_array_almost_equal(img, expected)
def test_imread_multipage_rgb_tif():
img = imread(os.path.join(data_dir, 'multipage_rgb.tif'))
assert img.shape == (2, 10, 10, 3), img.shape
class TestSave:
def roundtrip(self, dtype, x):
f = NamedTemporaryFile(suffix='.tif')
fname = f.name
f.close()
sio.imsave(fname, x)
y = sio.imread(fname)
imsave(fname, x)
y = imread(fname)
assert_array_equal(x, y)
def test_imsave_roundtrip(self):