def test_two_most_pleasure(self):
additive = Strategies(test_data.movie_set_2, test_data.data_set_2)
movie = additive.most_pleasure()
max_score = additive.maximum_values()
self.assertEqual(max_score, [7, 5])
self.assertEqual(movie, "A")
def test_one_most_pleasure(self):
additive = Strategies(test_data.movie_set_1, test_data.data_set_1)
movie = additive.most_pleasure()
max_score = additive.maximum_values()
self.assertEqual(max_score, [9, 10, 10])
self.assertEqual(movie, ['B', 'C'])