def test_completed_step(self): """ This tests the functionality of the method completed_step. """ global nsteps nsteps = 0 f = lambda t, x: x**0.25 def completed_step(solver): global nsteps nsteps += 1 mod = Explicit_Problem(f, 1.0) mod.step_events = completed_step sim = CVode(mod) sim.simulate(2., 100) assert len(sim.t_sol) == 101 assert nsteps == sim.statistics["nsteps"] sim = CVode(mod) nsteps = 0 sim.simulate(2.) assert len(sim.t_sol) == sim.statistics["nsteps"] + 1 assert nsteps == sim.statistics["nsteps"]
def test_completed_step(self): """ This tests the functionality of the method completed_step. """ global nsteps nsteps = 0 f = lambda t,x: x**0.25 def completed_step(solver): global nsteps nsteps += 1 mod = Explicit_Problem(f, 1.0) mod.step_events = completed_step sim = CVode(mod) sim.simulate(2., 100) assert len(sim.t_sol) == 101 assert nsteps == sim.statistics["nsteps"] sim = CVode(mod) nsteps = 0 sim.simulate(2.) assert len(sim.t_sol) == sim.statistics["nsteps"]+1 assert nsteps == sim.statistics["nsteps"]