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)