def test_num_per_timestep(self):
'test ContinuousRelease when a fixed rate per timestep is given'
r = ContinuousRelease(self.rel_time,
self.pos,
initial_elements=1000,
num_per_timestep=100)
assert r.num_elements is None
assert r.num_elements_to_release(self.rel_time, 100) == 1100
r.initial_done = True
for ts in (200, 400):
num = r.num_elements_to_release(self.rel_time, ts)
assert num == 100
def test_num_per_timestep(self):
'test ContinuousRelease when a fixed rate per timestep is given'
r = ContinuousRelease(self.rel_time,
self.pos,
initial_elements=1000,
num_per_timestep=100)
assert r.num_elements is None
assert r.num_elements_to_release(self.rel_time, 100) == 1100
r.initial_done = True
for ts in (200, 400):
num = r.num_elements_to_release(self.rel_time, ts)
assert num == 100
def test_property_num_per_timestep_elements(self):
'''
test either num_elements or num_per_timestep is set but not both
also test the num_elements_to_release references correct method
'''
r = ContinuousRelease(self.rel_time,
self.pos,
initial_elements=100,
num_per_timestep=100)
r.num_elements = 10
assert r.num_per_timestep is None
assert r.num_elements_to_release(self.rel_time, 900) == 110
r.num_per_timestep = 100
assert r.num_elements is None
assert r.num_elements_to_release(self.rel_time, 900) == 200
def test_property_num_per_timestep_elements(self):
'''
test either num_elements or num_per_timestep is set but not both
also test the num_elements_to_release references correct method
'''
r = ContinuousRelease(self.rel_time,
self.pos,
initial_elements=100,
num_per_timestep=100)
r.num_elements = 10
assert r.num_per_timestep is None
assert r.num_elements_to_release(self.rel_time, 900) == 110
r.num_per_timestep = 100
assert r.num_elements is None
assert r.num_elements_to_release(self.rel_time, 900) == 200
