Fixed database typo and removed unnecessary class identifier.

venv/Lib/site-packages/skimage/io/tests/test_plugin_util.py

@@ -0,0 +1,69 @@
+import numpy as np
+from skimage.io._plugins.util import prepare_for_display, WindowManager
+
+from skimage._shared import testing
+from skimage._shared.testing import assert_array_equal, TestCase
+from skimage._shared._warnings import expected_warnings
+
+
+np.random.seed(0)
+
+
+class TestPrepareForDisplay(TestCase):
+    def test_basic(self):
+        prepare_for_display(np.random.rand(10, 10))
+
+    def test_dtype(self):
+        x = prepare_for_display(np.random.rand(10, 15))
+        assert x.dtype == np.dtype(np.uint8)
+
+    def test_grey(self):
+        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_color(self):
+        prepare_for_display(np.random.rand(10, 10, 3))
+
+    def test_alpha(self):
+        prepare_for_display(np.random.rand(10, 10, 4))
+
+    def test_wrong_dimensionality(self):
+        with testing.raises(ValueError):
+            prepare_for_display(np.random.rand(10, 10, 1, 1))
+
+    def test_wrong_depth(self):
+        with testing.raises(ValueError):
+            prepare_for_display(np.random.rand(10, 10, 5))
+
+
+class TestWindowManager(TestCase):
+    callback_called = False
+
+    @testing.fixture(autouse=True)
+    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')