mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-04 19:07:50 +08:00
Stefan van der Walt
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
from numpy.testing import *
|
|
|
|
from scikits.image import io
|
|
from scikits.image.io._plugins import plugin
|
|
|
|
from copy import deepcopy
|
|
|
|
def read(fname, as_grey=False, dtype=None):
|
|
assert fname == 'test.png'
|
|
assert as_grey == True
|
|
assert dtype == 'i4'
|
|
|
|
def save(fname, arr):
|
|
assert fname == 'test.png'
|
|
assert arr == [1, 2, 3]
|
|
|
|
def show(arr, plugin_arg=None):
|
|
assert arr == [1, 2, 3]
|
|
assert plugin_arg == (1, 2)
|
|
|
|
def show_other(arr):
|
|
return "other"
|
|
|
|
def setup_module(self):
|
|
self.backup_plugin_store = deepcopy(plugin.plugin_store)
|
|
plugin.register('testcase', read=read, save=save, show=show)
|
|
plugin.register('other', show=show_other)
|
|
|
|
def teardown_module(self):
|
|
plugin.plugin_store = self.backup_plugin_store
|
|
|
|
class TestPlugin:
|
|
def test_read(self):
|
|
io.imread('test.png', as_grey=True, dtype='i4', plugin='testcase')
|
|
|
|
def test_save(self):
|
|
io.imsave('test.png', [1, 2, 3], plugin='testcase')
|
|
|
|
def test_show(self):
|
|
io.imshow([1, 2, 3], plugin_arg=(1, 2), plugin='testcase')
|
|
|
|
def test_use(self):
|
|
plugin.use('other', 'show')
|
|
assert io.imshow(None) == 'other'
|
|
|
|
def test_available(self):
|
|
plugin.use('other', 'show')
|
|
d = plugin.available('show')
|
|
assert d['show'][0] == 'other'
|
|
|
|
def test_load(self):
|
|
plugin.load('test')
|
|
fname, arr = io.imsave('outfile', [1, 2, 3])
|
|
assert_equal(fname, 'outfile')
|
|
assert_equal(arr, [1, 2, 3])
|
|
assert_equal(plugin.available('save')['save'][0], 'test')
|
|
|
|
if __name__ == "__main__":
|
|
run_module_suite()
|