Spaces:
Running
Running
File size: 3,619 Bytes
ba2f5d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
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}
|