mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 13:54:41 +08:00
Steven Silvester
9e8f91930e
Start handling warnings in data, exposure, and draw Add a known_warning decorator and suppress warnings in color pkg Use the existing all_warnings context manager Raise warnings in data Raise warnings in draw Raise warnings in exposure Suppress warnings in exposure tests Add comments about warning suppressions Raise warnings in feature Fix warnings in filter package Add warning handling to graph Handle warnings in io package
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
from skimage.io._plugins.util import prepare_for_display, WindowManager
|
|
from skimage._shared.utils import all_warnings
|
|
|
|
from numpy.testing import *
|
|
import numpy as np
|
|
|
|
np.random.seed(0)
|
|
|
|
|
|
class TestPrepareForDisplay:
|
|
def test_basic(self):
|
|
with all_warnings(): # precision loss
|
|
prepare_for_display(np.random.rand(10, 10))
|
|
|
|
def test_dtype(self):
|
|
with all_warnings(): # precision loss
|
|
x = prepare_for_display(np.random.rand(10, 15))
|
|
assert x.dtype == np.dtype(np.uint8)
|
|
|
|
def test_grey(self):
|
|
with all_warnings(): # precision loss
|
|
tmp = np.arange(12, dtype=float).reshape((4, 3)) / 11
|
|
x = prepare_for_display(tmp)
|
|
assert_array_equal(x[..., 0], x[..., 2])
|
|
assert x[0, 0, 0] == 0
|
|
assert x[3, 2, 0] == 255
|
|
|
|
def test_colour(self):
|
|
with all_warnings(): # precision loss
|
|
prepare_for_display(np.random.rand(10, 10, 3))
|
|
|
|
def test_alpha(self):
|
|
with all_warnings(): # precision loss
|
|
prepare_for_display(np.random.rand(10, 10, 4))
|
|
|
|
@raises(ValueError)
|
|
def test_wrong_dimensionality(self):
|
|
with all_warnings(): # precision loss
|
|
prepare_for_display(np.random.rand(10, 10, 1, 1))
|
|
|
|
@raises(ValueError)
|
|
def test_wrong_depth(self):
|
|
with all_warnings(): # precision loss
|
|
prepare_for_display(np.random.rand(10, 10, 5))
|
|
|
|
|
|
class TestWindowManager:
|
|
callback_called = False
|
|
|
|
def setup(self):
|
|
self.wm = WindowManager()
|
|
self.wm.acquire('test')
|
|
|
|
def test_add_window(self):
|
|
self.wm.add_window('window1')
|
|
self.wm.remove_window('window1')
|
|
|
|
def callback(self):
|
|
self.callback_called = True
|
|
|
|
def test_callback(self):
|
|
self.wm.register_callback(self.callback)
|
|
self.wm.add_window('window')
|
|
self.wm.remove_window('window')
|
|
assert self.callback_called
|
|
|
|
def test_has_images(self):
|
|
assert not self.wm.has_windows()
|
|
self.wm.add_window('window')
|
|
assert self.wm.has_windows()
|
|
|
|
def teardown(self):
|
|
self.wm._release('test')
|
|
|
|
if __name__ == "__main__":
|
|
run_module_suite()
|