def test_accumulate_with_cross_reference_mirror():
class S1(System):
@accumulate
def a(self):
return self.b + 1
@accumulate
def b(self):
return self.a + 2
class S2(System):
@accumulate
def a(self):
return self.b + 2
@accumulate
def b(self):
return self.a + 1
s1 = instance(S1)
s2 = instance(S2)
c1 = s1.context
c2 = s2.context
assert s1.a == s2.b and s1.b == s2.a
c1.advance()
c2.advance()
assert s1.a == s2.b and s1.b == s2.a
c1.advance()
c2.advance()
assert s1.a == s2.b and s1.b == s2.a
c1.advance()
c2.advance()
assert s1.a == s2.b and s1.b == s2.a
def test_flag():
class S(System):
@flag
def a(self):
return True
@flag
def b(self):
return False
@flag(prob=0)
def c(self):
return True
@flag(prob=1)
def d(self):
return False
zero = derive(0)
one = derive(1)
@flag(prob='zero')
def e(self):
return True
@flag(prob='one')
def f(self):
return False
s = instance(S)
assert s.a and not s.b
assert not s.c and not s.d
assert not s.e and not s.f
def test_args():
class S(System):
@derive(alias='a')
def a_long_named_var(self):
return 1
@derive
def b(self):
return self.a_long_named_var
@derive
def c(self, a_long_named_var):
return a_long_named_var
@derive
def d(self, a):
return a
@derive
def e(self, ee='a'):
return ee
@derive
def f(self, ff=1):
return ff
s = instance(S)
assert s.a == s.b == s.c == s.d == s.e == s.f == 1
def test_accumulate_distribute():
class S(System):
@drive
def s(self):
return {'s': self.context.time * 100}
@accumulate
def d1(self):
return self.s * 0.2
@accumulate
def d2(self):
return self.s * 0.3
@accumulate
def d3(self):
return self.s * 0.5
s = instance(S)
c = s.context
assert c.time == 0 and s.s == 0 and s.d1 == 0 and s.d2 == 0 and s.d3 == 0
c.advance()
assert c.time == 1 and s.s == 100 and s.d1 == 0 and s.d2 == 0 and s.d3 == 0
c.advance()
assert c.time == 2 and s.s == 200 and s.d1 == 20 and s.d2 == 30 and s.d3 == 50
c.advance()
assert c.time == 3 and s.s == 300 and s.d1 == 60 and s.d2 == 90 and s.d3 == 150
c.advance()
assert c.time == 4 and s.s == 400 and s.d1 == 120 and s.d2 == 180 and s.d3 == 300
def test_produce_with_kwargs():
class S(System):
@produce
def a(self):
return (S, {
't': self.context.time + 1
}) # produced in the next time step
@constant(init=0)
def t(self):
return None
s = instance(S)
c = s.context
assert len(s.children) == 0 and s.t == 0
c.advance()
assert len(s.children) == 1 and s.t == 0
s1 = s.children[0]
assert len(s1.children) == 0 and s1.t == 1
c.advance()
s2 = s.children[1]
s11 = s1.children[0]
assert len(s.children) == 2 and len(s1.children) == 1 and len(
s2.children) == 0
assert s.t == 0 and s1.t == 1 and s2.t == 2 and s11.t == 2
def test_flip():
class S(System):
@derive
def a(self):
return 1
@accumulate
def b(self):
return self.a + 1
@flip
def sa(self):
return self.a
@flip
def sb(self):
return self.b
s = instance(S)
c = s.context
assert s.a == 1 and s.b == 0
assert s.sa == 1 and s.sb == 0
c.advance()
assert s.a == 1 and s.b == 2
assert s.sa == 0 and s.sb == 2
c.advance()
assert s.a == 1 and s.b == 4
assert s.sa == 0 and s.sb == 4
def test_drive_with_system():
class T(System):
@derive
def a(self):
return 1
@derive
def b(self):
return 2
class S(System):
@system
def t(self):
return T
@drive(alias='aa')
def a(self):
return self.t
b = drive('t', alias='bb')
c = derive('t.b')
s = instance(S)
assert s.a == s.aa == s.t.a == 1
assert s.b == s.bb == s.t.b == s.c == 2
def test_inline():
class S(System):
a = derive(1)
b = derive(2, alias='c')
s = instance(S)
assert s.a == 1
assert s.b == s.c == 2
def test_getitem():
class S(System):
@derive
def a(self):
return 1
s = instance(S)
assert s.a == s['a'] == 1
def test_args_with_config():
class S(System):
@derive
def a(self, b):
return b
s = instance(S, config={'S': {'a': {'b': 1}}})
assert s.a == 1
def test_drive_with_key():
class S(System):
@drive(key='b')
def a(self):
return {'b': 1}
s = instance(S)
c = s.context
assert s.a == 1
def test_clock_with_config():
class S(System):
pass
s = instance(S, config={'Clock': {'start': 5, 'interval': 10}})
c = s.context
assert c.time == 5
c.advance()
assert c.time == 15
c.advance()
assert c.time == 25
def test_clock():
class S(System):
pass
s = instance(S)
c = s.context
assert c.time == 0
c.advance()
assert c.time == 1
c.advance()
assert c.time == 2
def test_produce_with_none():
class S(System):
@produce
def a(self):
return None
s = instance(S)
c = s.context
assert len(s.children) == 0
c.advance()
assert len(s.children) == 0
def test_systemproxy():
class T(System):
@derive
def a(self):
return 1
class S(System):
t = systemproxy(T)
s = instance(S)
assert s.a == s.t.a == 1
def test_parameter_with_config():
class S(System):
@parameter
def a(self):
return 1
s = instance(S, config={'S': {'a': 2}})
c = s.context
assert s.a == 2
c.advance()
assert s.a == 2
def test_parameter():
class S(System):
@parameter
def a(self):
return 1
s = instance(S)
c = s.context
assert s.a == 1
c.advance()
assert s.a == 1
def test_derive_with_cross_reference():
class S(System):
@derive
def a(self):
return self.b
@derive
def b(self):
return self.a
with pytest.raises(RecursionError):
s = instance(S)
def test_parameter_with_config_alias():
class S(System):
@parameter(alias='aa')
def a(self):
return 1
@parameter(alias='b')
def bb(self):
return 1
s = instance(S, config={'S': {'a': 2, 'b': 2}})
assert s.a == 2
assert s.b == 2
def test_parameter():
class S(System):
a = parameter(1, unit='m')
b = parameter(1, unit='m')
c = parameter(1, unit='m')
d = parameter(1, unit='m')
e = parameter(1)
@parameter(unit='m')
def f(self, ff='1m'):
return self.a + ff
s = instance(S, config={'S': {'a': 2, 'b': '2', 'c': '2m', 'd': '200cm', 'e': '2m'}})
assert s.a == s.b == s.c == s.d == s.e == U(2, 'm')
assert s.f == U(3, 'm')
def test_nounit():
class S(System):
@derive(unit='m')
def a(self):
return 1
@derive(nounit='a')
def b(self, a):
return a
s = instance(S)
assert isinstance(s.a, U.registry.Quantity)
assert s.a == U(1, 'm')
assert not isinstance(s.b, U.registry.Quantity)
assert s.b == 1
def test_nounit_with_alias():
class S(System):
@derive(alias='aa', unit='m')
def a(self):
return 1
@derive(alias='bb', nounit='aa')
def b(self, aa):
return aa
s = instance(S)
assert isinstance(s.aa, U.registry.Quantity)
assert s.aa == U(1, 'm')
assert not isinstance(s.bb, U.registry.Quantity)
assert s.bb == 1
def test_unit():
class S(System):
@derive(unit='m')
def a(self):
return 2
@derive(unit='s')
def b(self):
return 1
@derive(unit='m/s')
def c(self):
return self.a / self.b
s = instance(S)
assert s.a == U(2, 'm') and s.b == U(1, 's') and s.c == U(2, 'm/s')
def test_lotka_volterra(tmp_path):
class S(System):
@parameter(alias='a')
def prey_birth_rate(self):
return 1.0
@parameter(alias='b')
def prey_death_rate(self):
return 0.1
@parameter(alias='c')
def predator_death_rate(self):
return 1.5
@parameter(alias='d')
def predator_reproduction_rate(self):
return 0.75
@parameter(alias='H0')
def prey_initial_population(self):
return 10
@parameter(alias='P0')
def predator_initial_population(self):
return 5
@accumulate(alias='H', init='H0')
def prey_population(self, a, b, H, P):
return a*H - b*H*P
@accumulate(alias='P', init='P0')
def predator_population(self, b, c, d, H, P):
return d*b*H*P - c*P
s = instance(S, config={'Clock': {'interval': 0.01}})
c = s.context
T = range(2000)
H = []
P = []
for t in T:
c.advance()
#print(f't = {t}: H = {s.prey_population}, P={s.predator_population}')
H.append(s.prey_population)
P.append(s.predator_population)
plt.plot(T, H, label='Prey')
plt.plot(T, P, label='Predator')
plt.xlabel('Time')
plt.ylabel('Population')
plt.legend()
#plt.show()
plt.savefig(tmp_path/'lotka_volterra.png')
def test_produce():
class S(System):
@produce
def a(self):
return S
s = instance(S)
c = s.context
assert len(s.children) == 0
c.advance()
assert len(s.children) == 1 and len(s.children[0].children) == 0
c.advance()
assert len(s.children) == 2 and len(s.children[0].children) == 1 and len(
s.children[1].children) == 0
def test_drive_with_dict():
class S(System):
@drive
def a(self):
return {'a': self.context.time * 10}
s = instance(S)
c = s.context
assert c.time == 0 and s.a == 0
c.advance()
assert c.time == 1 and s.a == 10
c.advance()
assert c.time == 2 and s.a == 20
c.advance()
assert c.time == 3 and s.a == 30
def test_args_partial():
class S(System):
@derive
def a(self, b, c, d=0):
return b + c + d
@derive
def b(self):
return 1
s = instance(S)
assert s.a(c=1) == 2
assert s.a(c=2) == 3
assert s.a(c=2, d=1) == 4
assert s.a(1) == 2
def test_derive():
class S(System):
@derive
def a(self):
return 1
@derive
def b(self):
return 2
@derive
def c(self):
return self.a + self.b
s = instance(S)
assert s.a == 1 and s.b == 2 and s.c == 3
def test_optimize_with_unit():
class S(System):
@derive(unit='m')
def a(self):
return 2 * self.x
@derive(unit='m')
def b(self):
return self.x + U(1, 'm')
@optimize(lower=U(0, 'm'), upper=U(2, 'm'), unit='m')
def x(self):
return self.a - self.b
s = instance(S)
assert s.x == U(1, 'm')
assert s.a == s.b == U(2, 'm')
def test_optimize():
class S(System):
@derive
def a(self):
return 2 * self.x
@derive
def b(self):
return self.x + 1
@optimize(lower=0, upper=2)
def x(self):
return self.a - self.b
s = instance(S)
assert s.x == 1
assert s.a == s.b == 2
