def test_find_happiest_arrangement(self): """Tests find_happiest_arrangment().""" happiness, people = puzzle13.load_input_file('../test_input.txt') arrangements = puzzle13.possible_seating_arrangements(people) scored_arrangements = puzzle13.score_seating_arrangements(arrangements, \ happiness) happiest_arrangement = puzzle13.find_happiest_arrangement(scored_arrangements) self.assertEquals(330, scored_arrangements[happiest_arrangement])
def test_load_input_file(self): """Tests for load_input_file()""" happiness, people = puzzle13.load_input_file('../test_input.txt') self.assertEquals(54, happiness[('Alice', 'Bob')]) self.assertEquals(-7, happiness[('Bob', 'Carol')]) self.assertEquals(4, len(people)) self.assertTrue('Alice' in people) self.assertTrue('Bob' in people) self.assertTrue('Carol' in people) self.assertTrue('David' in people)