mirror of
https://github.com/gnuradio/gnuradio-companion.git
synced 2025-12-11 04:34:56 -06:00
107 lines
2.3 KiB
Python
107 lines
2.3 KiB
Python
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
|
|
a.foo = u'${ 10 + 2 }'
|
|
assert a.foo == 12
|
|
|
|
|
|
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'
|