mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-29 03:37:34 +08:00
Steven Silvester
78 lines
2.0 KiB
Python
78 lines
2.0 KiB
Python
|
|
from skimage import data
|
|
from skimage.viewer.qt import QtGui, QtCore
|
|
from skimage.viewer import ImageViewer, CollectionViewer, viewer_available
|
|
from skimage.transform import pyramid_gaussian
|
|
from skimage.viewer.plugins import OverlayPlugin
|
|
from skimage.filter import sobel
|
|
from numpy.testing import assert_equal
|
|
from numpy.testing.decorators import skipif
|
|
from skimage._shared.version_requirements import is_installed
|
|
|
|
|
|
@skipif(not viewer_available)
|
|
def test_viewer():
|
|
lena = data.lena()
|
|
coins = data.coins()
|
|
|
|
view = ImageViewer(lena)
|
|
import tempfile
|
|
_, filename = tempfile.mkstemp(suffix='.png')
|
|
|
|
view.show(False)
|
|
view.close()
|
|
view.save_to_file(filename)
|
|
view.open_file(filename)
|
|
assert_equal(view.image, lena)
|
|
view.image = coins
|
|
assert_equal(view.image, coins),
|
|
view.save_to_file(filename),
|
|
view.open_file(filename),
|
|
view.reset_image(),
|
|
assert_equal(view.image, coins)
|
|
|
|
|
|
def make_key_event(key):
|
|
return QtGui.QKeyEvent(QtCore.QEvent.KeyPress, key,
|
|
QtCore.Qt.NoModifier)
|
|
|
|
|
|
@skipif(not viewer_available)
|
|
def test_collection_viewer():
|
|
|
|
img = data.lena()
|
|
img_collection = tuple(pyramid_gaussian(img))
|
|
|
|
view = CollectionViewer(img_collection)
|
|
make_key_event(48)
|
|
|
|
view.update_index('', 2),
|
|
assert_equal(view.image, img_collection[2])
|
|
view.keyPressEvent(make_key_event(53))
|
|
assert_equal(view.image, img_collection[5])
|
|
view._format_coord(10, 10)
|
|
|
|
|
|
@skipif(not viewer_available)
|
|
@skipif(not is_installed('matplotlib', '>=1.2'))
|
|
def test_viewer_with_overlay():
|
|
img = data.coins()
|
|
ov = OverlayPlugin(image_filter=sobel)
|
|
viewer = ImageViewer(img)
|
|
viewer += ov
|
|
|
|
import tempfile
|
|
_, filename = tempfile.mkstemp(suffix='.png')
|
|
|
|
ov.color = 3
|
|
assert_equal(ov.color, 'yellow')
|
|
viewer.save_to_file(filename)
|
|
ov.display_filtered_image(img)
|
|
assert_equal(ov.overlay, img)
|
|
ov.overlay = None
|
|
assert_equal(ov.overlay, None)
|
|
ov.overlay = img
|
|
assert_equal(ov.overlay, img)
|
|
assert_equal(ov.filtered_image, img)
|
|
|