def test_finds_best_arrangement_with_noncommittal_guest(self): guests = parse_input(self.data) guests['Me'] = {} arrangement, happiness = find_best_arrangement(guests) self.assertEquals(len(arrangement), 5) self.assertEquals(happiness, 286)
def test_finds_best_arrangement(self): guests = parse_input(self.data) arrangement, happiness = find_best_arrangement(guests) self.assertEquals(len(arrangement), 4) self.assertEquals(happiness, 330)