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
|
import collections
import numbers
from grc.core.utils.descriptors import Evaluated, EvaluatedEnum, EvaluatedPInt
class A(object):
def __init__(self, **kwargs):
self.called = collections.defaultdict(int)
self.errors = []
self.namespace = kwargs
def add_error_message(self, msg):
self.errors.append(msg)
@property
def parent_block(self):
return self
def evaluate(self, expr):
self.called['evaluate'] += 1
return eval(expr, self.namespace)
@Evaluated(int, 1)
def foo(self):
self.called['foo'] += 1
return eval(self._foo)
bar = Evaluated(numbers.Real, 1.0, name='bar')
test = EvaluatedEnum(['a', 'b'], 'a', name='test')
lala = EvaluatedPInt()
def test_fixed_value():
a = A()
a.foo = 10
assert not hasattr(a, '_foo')
assert a.foo == 10
assert a.called['foo'] == 0
delattr(a, 'foo')
assert a.foo == 10
assert a.called['foo'] == 0
def test_evaled():
a = A()
a.foo = '${ 10 + 1 }'
assert getattr(a, '_foo') == '10 + 1'
assert a.foo == 11 and a.foo == 11
assert a.called['foo'] == 1
assert a.called['evaluate'] == 0
delattr(a, 'foo')
assert a.foo == 11 and a.foo == 11
assert a.called['foo'] == 2
assert not a.errors
def test_evaled_with_default():
a = A()
a.bar = '${ 10 + 1 }'
assert getattr(a, '_bar') == '10 + 1'
assert a.bar == 11.0 and type(a.bar) == int
assert a.called['evaluate'] == 1
assert not a.errors
def test_evaled_int_with_default():
a = A(ll=10)
a.lala = '${ ll * 2 }'
assert a.lala == 20
a.namespace['ll'] = -10
assert a.lala == 20
del a.lala
assert a.lala == 1
assert not a.errors
def test_evaled_enum_fixed_value():
a = A()
a.test = 'a'
assert not hasattr(a, '_test')
assert a.test == 'a' and type(a.test) == str
assert not a.errors
def test_evaled_enum():
a = A(bla=False)
a.test = '${ "a" if bla else "b" }'
assert a.test == 'b'
a.namespace['bla'] = True
assert a.test == 'b'
del a.test
assert a.test == 'a'
assert not a.errors
def test_class_access():
a = A()
a.foo = '${ meme }'
descriptor = getattr(a.__class__, 'foo')
assert descriptor.name_raw == '_foo'
|
