def test_two_average_without_misery(self):
additive = Strategies(test_data.movie_set_2, test_data.data_set_2)
movie = additive.average_without_misery()
max_score = additive.average_values()
self.assertEqual(max_score, [0, 9])
self.assertEqual(movie, "B")
def test_one_average_without_misery(self):
additive = Strategies(test_data.movie_set_1, test_data.data_set_1)
movie = additive.average_without_misery()
max_score = additive.average_values()
self.assertEqual(max_score, [22, 24, 21])
self.assertEqual(movie, 'B')
def test_five_average_without_misery(self):
additive = Strategies(test_data.movie_set_5_6, test_data.data_set_5)
movie = additive.average_without_misery()
max_score = additive.average_values()
self.assertEqual(max_score, [20, 0, 21, 25, 26, 28, 0, 15, 0, 23])
self.assertEqual(movie, 'F')
def test_six_average_without_misery(self):
additive = Strategies(test_data.movie_set_5_6, test_data.data_set_6)
movie = additive.average_without_misery()
max_score = additive.average_values()
self.assertEqual(max_score, [0, 18, 0, 22, 26, 26, 17, 23, 0, 22])
self.assertEqual(movie, ['E', 'F'])
def test_four_average_without_misery(self):
additive = Strategies(test_data.movie_set_4, test_data.data_set_4)
movie = additive.average_without_misery()
max_score = additive.average_values()
self.assertEqual(max_score, [0, 0, 0, 0])
self.assertEqual(movie, ["A", "B", "C", "D"])