def test_save_to_directory_and_load(self): self.ODE = jitcode(**self.argdict) self.ODE.compile_C(modulename=modulename) self.ODE.save_compiled(self.tmpfile(""), overwrite=True) self.ODE = ode_from_module_file(self.tmpfile(self.filename)) self.ODE.set_integrator('dopri5') self.ODE.set_initial_value(y0,0.0) self.assertTrue(_is_C(self.ODE.f)) self.assertIsNone(self.ODE.jac)
def test_save_and_load_from_different_directory(self): self.ODE = jitcode(wants_jacobian=True, **self.argdict) self.ODE.save_compiled(self.filename, overwrite=True) shutil.move(self.filename,self.tmpfile(self.filename)) self.ODE = ode_from_module_file(self.tmpfile(self.filename)) self.ODE.set_integrator('lsoda') self.ODE.set_initial_value(y0,0.0) self.assertTrue(_is_C(self.ODE.f)) self.assertTrue(_is_C(self.ODE.jac))
def test_save_with_default_name_and_load(self): self.ODE = jitcode(wants_jacobian=True, **self.argdict) self.ODE.save_compiled("", overwrite=True) self.filename = self.ODE._modulename + ".so" shutil.move(self.filename,self.tmpfile(self.filename)) self.ODE = ode_from_module_file(self.tmpfile(self.filename)) self.ODE.set_integrator('lsoda') self.ODE.set_initial_value(y0,0.0) self.assertTrue(_is_C(self.ODE.f)) self.assertTrue(_is_C(self.ODE.jac))
def test_save_and_load_with_jac(self): self.ODE = jitcode(wants_jacobian=True, **self.argdict) self.ODE.save_compiled(self.filename, overwrite=True) target = os.path.join(self.directory,self.filename) shutil.move(self.filename,target) self.ODE = ode_from_module_file(target) self.ODE.set_integrator('dopri5') self.ODE.set_initial_value(y0,0.0) self.assertTrue(_is_C(self.ODE.f)) self.assertTrue(_is_C(self.ODE.jac))