def test_interface(self):
rk = RKSolver(0, 1)
rk.initial_time = 0.5
rk.final_time = 5
rk.n_steps = 10
self.assertEqual(rk.initial_time,0.5)
self.assertEqual(rk.final_time,5)
self.assertEqual(rk.n_steps,10)
eqn = Seir()
rk.equation = eqn
self.assertEqual(rk.equation, eqn)
rk.output_frequency = 20
self.assertEqual(rk.output_frequency, 20)
def test_correct_caching(self):
# setup equation
eqn = Seir(population=1)
# setup ode solver
ti = 0.
tf = 2.
n_steps = 3
rk = RKSolver(ti, tf, n_steps)
rk.output_frequency = 1
rk.set_output_storing_flag(True)
rk.equation = eqn
u0 = np.array([100., 0., 10., 0.])
du0_dp = np.zeros((eqn.n_components(), eqn.n_parameters()))
rk.set_initial_condition(u0, du0_dp)
rk.set_output_gradient_flag(True)
# setup cached simulation object
cached_sim = CachedSEIRSimulation(rk)
params = np.array([2.3, 0.2, 1. / 3., 1. / 4.])
(f, df) = cached_sim(params)
f1 = np.copy(f)
df1 = np.copy(df)
params2 = np.array([2.32, 0.2, 1. / 3., 1. / 4.])
(f2, df2) = cached_sim(params2)
assert not np.allclose(f1, f2)
assert not np.allclose(df1, df2)
rk.final_time = 3
(f3, df3) = cached_sim(params)
assert not np.allclose(f1, f3)
assert not np.allclose(df1, df3)