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
|
import pytest
from app_model.expressions._context_keys import (
ContextKey,
ContextKeyInfo,
ContextNamespace,
)
def test_context_key_info() -> None:
key = ContextKey("default", "description", None, id="some_key")
info = ContextKey.info()
assert isinstance(info, list) and len(info)
assert all(isinstance(x, ContextKeyInfo) for x in info)
assert "some_key" in {x.key for x in info}
assert repr(key) == "Expr.parse('some_key')"
assert repr(key == 1) == "Expr.parse('some_key == 1')"
def _adder(x: list) -> int:
return sum(x)
def test_context_namespace() -> None:
class Ns(ContextNamespace):
my_key = ContextKey[list, int](0, "description", _adder)
optional_key = ContextKey[None, str](description="might be missing")
assert "my_key" in Ns.__members__
assert str(Ns.my_key) == "my_key"
assert any(x.description == "description" for x in ContextKey.info())
# make sure the type hints were inferred from adder
assert Ns.my_key.__orig_class__.__args__ == (list, int) # type: ignore
assert isinstance(Ns.my_key, ContextKey)
ctx: dict = {}
ns = Ns(ctx)
assert ns.my_key == 0
assert ctx["my_key"] == 0
ns.my_key = 2
assert ctx["my_key"] == 2
assert "optional_key" not in ctx
assert ns.optional_key is ContextKey.MISSING
ns.reset("optional_key") # shouldn't raise error to reset a missing key
# maybe the key is there though
ctx["optional_key"] = "hi"
assert ns.optional_key == "hi"
ns.reset_all()
assert ctx["my_key"] == 0
assert "optional_key" not in ctx
assert repr(ns) == "{'my_key': 0, 'optional_key': MISSING}"
assert Ns.my_key.eval(ctx) == 0
def test_good_naming() -> None:
with pytest.raises(RuntimeError):
# you're not allowed to create a key with an id different from
# it's attribute name
class Ns(ContextNamespace):
my_key = ContextKey(id="not_my_key") # type: ignore
|
