mirror of
https://github.com/aykhans/AzSuicideDataVisualization.git
synced 2025-04-22 10:28:02 +00:00
124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
from ..plugin_registry import PluginRegistry
|
|
from typing import Callable
|
|
|
|
|
|
class TypedCallableRegistry(PluginRegistry[Callable[[int], int]]):
|
|
pass
|
|
|
|
|
|
class GeneralCallableRegistry(PluginRegistry):
|
|
_global_settings = {"global_setting": None}
|
|
|
|
@property
|
|
def global_setting(self):
|
|
return self._global_settings["global_setting"]
|
|
|
|
@global_setting.setter
|
|
def global_setting(self, val):
|
|
self._global_settings["global_setting"] = val
|
|
|
|
|
|
def test_plugin_registry():
|
|
plugins = TypedCallableRegistry()
|
|
|
|
assert plugins.names() == []
|
|
assert plugins.active == ""
|
|
assert plugins.get() is None
|
|
assert repr(plugins) == "TypedCallableRegistry(active='', registered=[])"
|
|
|
|
plugins.register("new_plugin", lambda x: x ** 2)
|
|
assert plugins.names() == ["new_plugin"]
|
|
assert plugins.active == ""
|
|
assert plugins.get() is None
|
|
assert repr(plugins) == (
|
|
"TypedCallableRegistry(active='', " "registered=['new_plugin'])"
|
|
)
|
|
|
|
plugins.enable("new_plugin")
|
|
assert plugins.names() == ["new_plugin"]
|
|
assert plugins.active == "new_plugin"
|
|
assert plugins.get()(3) == 9
|
|
assert repr(plugins) == (
|
|
"TypedCallableRegistry(active='new_plugin', " "registered=['new_plugin'])"
|
|
)
|
|
|
|
|
|
def test_plugin_registry_extra_options():
|
|
plugins = GeneralCallableRegistry()
|
|
|
|
plugins.register("metadata_plugin", lambda x, p=2: x ** p)
|
|
plugins.enable("metadata_plugin")
|
|
assert plugins.get()(3) == 9
|
|
|
|
plugins.enable("metadata_plugin", p=3)
|
|
assert plugins.active == "metadata_plugin"
|
|
assert plugins.get()(3) == 27
|
|
|
|
# enabling without changing name
|
|
plugins.enable(p=2)
|
|
assert plugins.active == "metadata_plugin"
|
|
assert plugins.get()(3) == 9
|
|
|
|
|
|
def test_plugin_registry_global_settings():
|
|
plugins = GeneralCallableRegistry()
|
|
|
|
# we need some default plugin, but we won't do anything with it
|
|
plugins.register("default", lambda x: x)
|
|
plugins.enable("default")
|
|
|
|
# default value of the global flag
|
|
assert plugins.global_setting is None
|
|
|
|
# enabling changes the global state, not the options
|
|
plugins.enable(global_setting=True)
|
|
assert plugins.global_setting is True
|
|
assert plugins._options == {}
|
|
|
|
# context manager changes global state temporarily
|
|
with plugins.enable(global_setting="temp"):
|
|
assert plugins.global_setting == "temp"
|
|
assert plugins._options == {}
|
|
assert plugins.global_setting is True
|
|
assert plugins._options == {}
|
|
|
|
|
|
def test_plugin_registry_context():
|
|
plugins = GeneralCallableRegistry()
|
|
|
|
plugins.register("default", lambda x, p=2: x ** p)
|
|
|
|
# At first there is no plugin enabled
|
|
assert plugins.active == ""
|
|
assert plugins.options == {}
|
|
|
|
# Make sure the context is set and reset correctly
|
|
with plugins.enable("default", p=6):
|
|
assert plugins.active == "default"
|
|
assert plugins.options == {"p": 6}
|
|
|
|
assert plugins.active == ""
|
|
assert plugins.options == {}
|
|
|
|
# Make sure the context is reset even if there is an error
|
|
try:
|
|
with plugins.enable("default", p=6):
|
|
assert plugins.active == "default"
|
|
assert plugins.options == {"p": 6}
|
|
raise ValueError()
|
|
except ValueError:
|
|
pass
|
|
|
|
assert plugins.active == ""
|
|
assert plugins.options == {}
|
|
|
|
# Enabling without specifying name uses current name
|
|
plugins.enable("default", p=2)
|
|
|
|
with plugins.enable(p=6):
|
|
assert plugins.active == "default"
|
|
assert plugins.options == {"p": 6}
|
|
|
|
assert plugins.active == "default"
|
|
assert plugins.options == {"p": 2}
|