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
