def test_save_load_midrun_no_movers(images_dir, uncertain): """ create model, save it after 1step, then load and check equality of original model and persisted model Remove all movers and ensure it still works as expected """ model = make_model(images_dir, uncertain) for mover in model.movers: del model.movers[mover.id] model.step() print 'saving scnario ..' scene = Scenario(saveloc_, model) scene.save() scene.model = None # make it none - load from persistence print 'loading scenario ..' model2 = scene.load() for sc in zip(model.spills.items(), model2.spills.items()): # need to change both atol since reading persisted data sc[0]._array_allclose_atol = 1e-5 sc[1]._array_allclose_atol = 1e-5 assert model.spills == model2.spills assert model == model2
def test_save_load_scenario(images_dir, uncertain): model = make_model(images_dir, uncertain) print 'saving scnario ..' scene = Scenario(saveloc_, model) scene.save() scene.model = None # make it none - load from persistence print 'loading scenario ..' model2 = scene.load() assert model == model2
def test_save_load_midrun_scenario(images_dir, uncertain): """ create model, save it after 1step, then load and check equality of original model and persisted model """ model = make_model(images_dir, uncertain) model.step() print 'saving scnario ..' scene = Scenario(saveloc_, model) scene.save() scene.model = None # make it none - load from persistence print 'loading scenario ..' model2 = scene.load() for sc in zip(model.spills.items(), model2.spills.items()): sc[0]._array_allclose_atol = 1e-5 # need to change both atol sc[1]._array_allclose_atol = 1e-5 assert model.spills == model2.spills assert model == model2