def test_clear_resolvers_and_has_resolver_legacy(
restore_resolvers: Any) -> None:
assert not OmegaConf.has_resolver("foo")
OmegaConf.legacy_register_resolver("foo", lambda x: int(x) + 10)
assert OmegaConf.has_resolver("foo")
OmegaConf.clear_resolvers()
assert not OmegaConf.has_resolver("foo")
def test_clear_resolver(restore_resolvers: Any, register_resolver_params: Any,
name: str, expected: Any) -> None:
if register_resolver_params:
OmegaConf.register_new_resolver(**register_resolver_params)
assert expected["pre_clear"] == OmegaConf.has_resolver(name)
assert OmegaConf.clear_resolver(name) == expected["result"]
assert not OmegaConf.has_resolver(name)
def load_config(path: Path) -> DictConfig:
if not OmegaConf.has_resolver("range"):
OmegaConf.register_resolver(
"range", lambda start, stop, step: range(int(start), int(stop), int(step))
)
cfg = OmegaConf.load(path)
return cast(DictConfig, cfg)
def get_test_configuration(config_override: list = []):
test_path = os.path.dirname(os.path.abspath(__file__))
source_path = os.path.join(test_path, 'data')
output_path = os.path.join(test_path, 'output')
if not os.path.exists(output_path):
os.makedirs(output_path)
if not OmegaConf.has_resolver('output_path'):
OmegaConf.register_new_resolver('output_path',
lambda sub_path: output_path)
OmegaConf.register_new_resolver('source_path',
lambda sub_path: source_path)
if not GlobalHydra().is_initialized():
hydra.initialize(config_path='data/config', caller_stack_depth=2)
return hydra.compose("test_config.yaml", overrides=config_override)
def test_clear_resolvers_and_has_resolver(restore_resolvers: Any) -> None:
assert not OmegaConf.has_resolver("foo")
OmegaConf.register_new_resolver("foo", lambda x: x + 10)
assert OmegaConf.has_resolver("foo")
OmegaConf.clear_resolvers()
assert not OmegaConf.has_resolver("foo")
