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")