def test_system_rule_variable():
ctx = Qit()
x = Variable(Int(), "x")
f = Function("f").takes(Int(), "a").returns(Int()).reads(x).code("return x;")
s = System(Int().values(5), (f,))
result = ctx.run(s.states(3).iterate(), args={"x": 5})
assert result == [5]
def test_system_in_product():
ctx = Qit()
f = Function("f").takes(Int(), "x").returns(Int()).code("return x * 10;")
g = Function("g").takes(Int(), "x").returns(Vector(Int())).code("return { x + 1, x + 2 };")
v = Int().values(10, 20)
s = System(v, (f,g))
D = s.states(2)
P = Product(D, D)
values = ctx.run(D.iterate().take(1000))
pairs = set(itertools.product(values, values))
result = ctx.run(P.iterate().take(1000))
assert len(result) == 484 # 22 * 22
assert pairs == set(result)
def test_system_basic():
ctx = Qit()
# Rules
f = Function("f").takes(Int(), "x").returns(Int()).code("return x * 10;")
g = Function("g").takes(Int(), "x").returns(Vector(Int()))
g.code("if(x % 2 == 0) return {}; else return { x + 1, x + 2 };")
h = Function("h").takes(Int(), "x").returns(Int()).code("return -x;")
# Initial states
v = Int().values(10, 21)
# System
s = System(v, (f, g, h))
result = ctx.run(s.states(2).iterate().sort())
expected = [-210, # 21, f, h
-100, # 10, f, h
-23, # 21, g, h
-22, # 21, g, h
-21, # 21, h
-20, # 20, h, g
-19, # 20, h, g
-10, # 10, h
10, # 10
21, # 21
22, # 21, g
23, # 21, g
24, # 21, g, g
25, # 21, g, g
100, # 10, f
210, # 21, f
220, # 21, g, f
230, # 21, g, f
1000,# 10, f, f
2100,# 21, f, f
]
assert result == expected
def test_system_empty():
s = System((Int() * Int()).values(), ())
result = Qit().run(s.states(10).iterate())
assert result == []
