def test_five_most_pleasure(self):
additive = Strategies(test_data.movie_set_5_6, test_data.data_set_5)
movie = additive.most_pleasure()
max_score = additive.maximum_values()
self.assertEqual(max_score, [8, 10, 8, 10, 9, 10, 10, 6, 7, 10])
self.assertEqual(movie, ['B', 'D', 'F', 'G', 'J'])
def test_six_most_pleasure(self):
additive = Strategies(test_data.movie_set_5_6, test_data.data_set_6)
movie = additive.most_pleasure()
max_score = additive.maximum_values()
self.assertEqual(max_score, [10, 9, 8, 9, 10, 9, 6, 9, 10, 8])
self.assertEqual(movie, ['A', 'E', 'I'])
def test_four_most_pleasure(self):
additive = Strategies(test_data.movie_set_4, test_data.data_set_4)
movie = additive.most_pleasure()
max_score = additive.maximum_values()
self.assertEqual(max_score, [10, 8, 9, 7])
self.assertEqual(movie, 'A')
def test_three_most_pleasure(self):
additive = Strategies(test_data.movie_set_3, test_data.data_set_3)
movie = additive.most_pleasure()
max_score = additive.maximum_values()
self.assertEqual(max_score, [1, 1, 1])
self.assertEqual(movie, ["A", "B", "C"])
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'])
def most_pleasure(self):
data = self.process_data()
strategies = Strategies(data[0], data[1])
return strategies.most_pleasure()