def setUp(self):
self.f = GRSpec(env_vars={"x"},
sys_vars={"y"},
env_init=["x"],
sys_safety=["y"],
env_prog=["!x", "x"],
sys_prog=["y&&!x"])
self.triv = GRSpec(env_vars=["x"],
sys_vars=["y"],
env_init=["x && !x"])
self.empty = GRSpec()
def test_str_to_int():
x = "a' = \"hehe\""
s = GRSpec(sys_vars={'a': ['hehe', 'haha']}, sys_safety=[x])
s.str_to_int()
assert x in s._ast
assert x in s._bool_int
print(s._bool_int[x])
assert s._bool_int[x] == "( ( X a ) = 0 )"
def test_replace_dependent_vars():
sys_vars = {'a': 'boolean', 'locA': (0, 4)}
sys_safe = ['!a', 'a & (locA = 3)']
spc = GRSpec(sys_vars=sys_vars, sys_safety=sys_safe)
bool2form = {'a': '(locA = 0) | (locA = 2)', 'b': '(locA = 1)'}
replace_dependent_vars(spc, bool2form)
correct_result = ('[](( ! ( ( locA = 0 ) | ( locA = 2 ) ) )) && '
'[](( ( ( locA = 0 ) | ( locA = 2 ) ) & ( locA = 3 ) ))')
assert str(spc) == correct_result
def test_GRSpec():
f = GRSpec(env_vars={'a'},
sys_vars={'b'},
env_init=['!a'],
sys_init=['!b'],
env_safety=['a -> X !a', '!a -> X a'],
sys_prog=['a && b'])
M = lily.synthesize(f)
assert M is not None
def test_declare_str_vars():
# declaring string-valued variables
g = GRSpec()
# neither int nor str values
with pytest.raises(TypeError):
g.declare(i=(0, 10, 'wrong'))
d = dict(name=['a', 'b', 'c', 'w'])
g.declare(**d)
assert g.sys_vars == d, g.sys_vars
assert g.env_vars == dict(), g.env_vars
def test_declare_int_vars():
# declaring integer-valued variables
g = GRSpec()
g.declare(i=[0, 10])
assert g.sys_vars == dict(i=(0, 10)), g.sys_vars
assert g.env_vars == dict(), g.env_vars
g.declare(j=(-5, 14), env=True)
assert g.sys_vars == dict(i=(0, 10)), g.sys_vars
assert g.env_vars == dict(j=(-5, 14)), g.env_vars
# attempt to redeclare "i" as env var
with pytest.raises(AssertionError):
g.declare(i=(0, 10), env=True)
def test_or(self):
g = GRSpec(env_vars={"z"}, env_prog=["!z"])
h = self.f | g
assert len(h.env_vars) == 2 and 'z' in h.env_vars
assert (len(h.env_prog) == len(self.f.env_prog) + 1
and '!z' in h.env_prog)
# Domain mismatch on system variable y
g.sys_vars = {"y": (0, 5)}
nt.assert_raises(ValueError, self.f.__or__, g)
# Domain mismatch on environment variable x
g.sys_vars = dict()
g.env_vars["x"] = (0, 3)
nt.assert_raises(ValueError, self.f.__or__, g)
def test_declare_boolean_vars():
# declaring Boolean-valued variables
g = GRSpec()
g.declare('a')
assert g.sys_vars == dict(a='boolean'), g.sys_vars
assert g.env_vars == dict(), g.env_vars
g.declare('a', 'b')
assert g.sys_vars == dict(a='boolean', b='boolean'), g.sys_vars
assert g.env_vars == dict(), g.env_vars
g.declare('c', env=True)
assert g.sys_vars == dict(a='boolean', b='boolean'), g.sys_vars
assert g.env_vars == dict(c='boolean'), g.env_vars
# attempt to redeclare "c" as sys var
with pytest.raises(AssertionError):
g.declare('c')
def test_compile_init():
env_vars = {'x': (0, 0), 'y': (0, 0), 'z': (0, 1)}
sys_vars = {'w': (0, 0)}
env_init = ['((((y = 0))) & (x = 0))']
sys_init = ['((w = 0))']
spc = GRSpec(env_vars=env_vars,
sys_vars=sys_vars,
env_init=env_init,
sys_init=sys_init)
code = spc.compile_init(no_str=True)
d = dict(x=0, y=0, z=0, w=0)
assert eval(code, d)
d = dict(x=0, y=1, z=1, w=0)
assert eval(code, d)
d = dict(x=0, y=0, z=0, w=1)
assert not eval(code, d)
