Exemple #1
0
def test_register_resolver_twice_error_legacy(restore_resolvers: Any) -> None:
    def foo() -> int:
        return 10

    OmegaConf.legacy_register_resolver("foo", foo)
    with pytest.raises(AssertionError):
        OmegaConf.register_new_resolver("foo", lambda: 10)
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")
Exemple #3
0
def test_resolver_dot_start_legacy(restore_resolvers: Any) -> None:
    OmegaConf.legacy_register_resolver("identity", lambda x: x)
    c = OmegaConf.create({
        "foo_nodot": "${identity:bar}",
        "foo_dot": "${identity:.bar}"
    })
    assert c.foo_nodot == "bar"
    assert c.foo_dot == ".bar"
def test_register_resolver_twice_error_legacy_and_regular(
    restore_resolvers: Any, ) -> None:
    def foo() -> int:
        return 10

    OmegaConf.legacy_register_resolver("foo", foo)
    with raises(ValueError):
        OmegaConf.register_new_resolver("foo", foo)
def test_resolver_cache_2_legacy(restore_resolvers: Any) -> None:
    OmegaConf.legacy_register_resolver("random", lambda _: random.randint(0, 10000000))
    c1 = OmegaConf.create({"k": "${random:_}"})
    c2 = OmegaConf.create({"k": "${random:_}"})

    assert c1.k != c2.k
    assert c1.k == c1.k
    assert c2.k == c2.k
Exemple #6
0
def test_resolver_that_allows_a_list_of_arguments_legacy(
        restore_resolvers: Any, resolver: Resolver, name: str, key: str,
        result: Any) -> None:
    OmegaConf.legacy_register_resolver("my_resolver", resolver)
    c = OmegaConf.create({name: key})
    assert c[name] == result
Exemple #7
0
def test_register_resolver_1_legacy(restore_resolvers: Any) -> None:
    OmegaConf.legacy_register_resolver("plus_10", lambda x: int(x) + 10)
    c = OmegaConf.create({"k": "${plus_10:990}"})

    assert type(c.k) == int
    assert c.k == 1000
Exemple #8
0
def test_clear_resolvers_legacy(restore_resolvers: Any) -> None:
    assert OmegaConf.get_resolver("foo") is None
    OmegaConf.legacy_register_resolver("foo", lambda x: int(x) + 10)
    assert OmegaConf.get_resolver("foo") is not None
    OmegaConf.clear_resolvers()
    assert OmegaConf.get_resolver("foo") is None