def test_load_pickle_household(person_crop_last_act, tmpdir): hh = Household('1') hh.add(person_crop_last_act) path = os.path.join(tmpdir, 'test.pkl') hh.pickle(path) loaded = load_pickle(path) assert loaded.people assert list(loaded.people['1'].attributes) == list( hh.people['1'].attributes)
def test_pickle_household(person_crop_last_act, tmpdir): hh = Household('1') hh.add(person_crop_last_act) path = os.path.join(tmpdir, 'test.pkl') hh.pickle(path) assert os.path.exists(path)