mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 02:30:48 +08:00
Tony S Yu
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
import os
|
|
|
|
from numpy.testing import assert_array_equal, raises, run_module_suite
|
|
import numpy as np
|
|
|
|
import skimage.io as io
|
|
from skimage.io.manage_plugins import plugin_store
|
|
from skimage import data_dir
|
|
|
|
|
|
def test_stack_basic():
|
|
x = np.arange(12).reshape(3, 4)
|
|
io.push(x)
|
|
|
|
assert_array_equal(io.pop(), x)
|
|
|
|
|
|
@raises(ValueError)
|
|
def test_stack_non_array():
|
|
io.push([[1, 2, 3]])
|
|
|
|
|
|
def test_imread_url():
|
|
# tweak data path so that file URI works on both unix and windows.
|
|
data_path = data_dir.lstrip(os.path.sep)
|
|
data_path = data_path.replace(os.path.sep, '/')
|
|
image_url = 'file:///{0}/camera.png'.format(data_path)
|
|
image = io.imread(image_url)
|
|
assert image.shape == (512, 512)
|
|
|
|
|
|
@raises(RuntimeError)
|
|
def test_imread_no_plugin():
|
|
# tweak data path so that file URI works on both unix and windows.
|
|
image_path = os.path.join(data_dir, 'lena.png')
|
|
plugins = plugin_store['imread']
|
|
plugin_store['imread'] = []
|
|
try:
|
|
io.imread(image_path)
|
|
finally:
|
|
plugin_store['imread'] = plugins
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_module_suite()
|