def test_invert_map(self):
input_dic = {0: 1, 1: 1, 2: 2}
result = invert_map(input_dic)
self.assertEqual(list(result.keys()), [1, 2])
self.assertEqual(result[1], [0, 1])
self.assertEqual(result[2], [2])
def build_house_individual_map(individual_house_map_arg):
# House -> List of individuals
return invert_map(individual_house_map_arg)
def build_workplace_individual_map(individual_workplace_map_arg):
# workplace -> Individuals
return invert_map(individual_workplace_map_arg)
def test_build_individual_houses_map__average_moroccan_household(self):
result = build_individual_houses_map(1000)
mean_family = np.mean([len(v) for k, v in invert_map(result).items()])
self.assertTrue(abs(mean_family - 4.52) < 0.3)