def test_pickle_population(person_crop_last_act, tmpdir):
population = Population()
hh = Household('1')
hh.add(person_crop_last_act)
population.add(hh)
path = os.path.join(tmpdir, 'test.pkl')
population.pickle(path)
assert os.path.exists(path)
def test_load_pickle_population(person_crop_last_act, tmpdir):
population = Population()
hh = Household('1')
hh.add(person_crop_last_act)
population.add(hh)
path = os.path.join(tmpdir, 'test.pkl')
population.pickle(path)
loaded = load_pickle(path)
assert loaded.households
assert list(loaded.households['1'].people) == list(
population.households['1'].people)