Fixed database typo and removed unnecessary class identifier.

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

@@ -0,0 +1,72 @@
+from contextlib import contextmanager
+
+from skimage import io
+from skimage.io import manage_plugins
+
+from skimage._shared import testing
+from skimage._shared.testing import assert_equal
+
+
+
+priority_plugin = 'pil'
+
+
+def setup():
+    io.use_plugin('pil')
+
+
+def teardown_module():
+    io.reset_plugins()
+
+
+@contextmanager
+def protect_preferred_plugins():
+    """Contexts where `preferred_plugins` can be modified w/o side-effects."""
+    preferred_plugins = manage_plugins.preferred_plugins.copy()
+    try:
+        yield
+    finally:
+        manage_plugins.preferred_plugins = preferred_plugins
+
+
+def test_failed_use():
+    with testing.raises(ValueError):
+        manage_plugins.use_plugin('asd')
+
+
+def test_use_priority():
+    manage_plugins.use_plugin(priority_plugin)
+    plug, func = manage_plugins.plugin_store['imread'][0]
+    assert_equal(plug, priority_plugin)
+
+    manage_plugins.use_plugin('matplotlib')
+    plug, func = manage_plugins.plugin_store['imread'][0]
+    assert_equal(plug, 'matplotlib')
+
+
+def test_load_preferred_plugins_all():
+    from skimage.io._plugins import pil_plugin, matplotlib_plugin
+
+    with protect_preferred_plugins():
+        manage_plugins.preferred_plugins = {'all': ['pil'],
+                                            'imshow': ['matplotlib']}
+        manage_plugins.reset_plugins()
+
+        for plugin_type in ('imread', 'imsave'):
+            plug, func = manage_plugins.plugin_store[plugin_type][0]
+            assert func == getattr(pil_plugin, plugin_type)
+        plug, func = manage_plugins.plugin_store['imshow'][0]
+        assert func == getattr(matplotlib_plugin, 'imshow')
+
+
+def test_load_preferred_plugins_imread():
+    from skimage.io._plugins import pil_plugin, matplotlib_plugin
+
+    with protect_preferred_plugins():
+        manage_plugins.preferred_plugins['imread'] = ['pil']
+        manage_plugins.reset_plugins()
+
+        plug, func = manage_plugins.plugin_store['imread'][0]
+        assert func == pil_plugin.imread
+        plug, func = manage_plugins.plugin_store['imshow'][0]
+        assert func == matplotlib_plugin.imshow, func.__module__