def test_best_seller(self): sellers = [ { "name": "Joaquina", "store": 2, "value": 1200.00 }, { "name": "Pedro", "store": 2, "value": 120.00 }, { "name": "Maria", "store": 1, "value": 450.00 }, { "name": "Fernanda", "store": 1, "value": 4000.00 }, { "name": "Patricia", "store": 1, "value": 100.00 }, ] rancking = SellersRanking() self.assertEqual(rancking.best_seller(sellers), ["Fernanda"])
class SellersRankingTests(unittest.TestCase): """Performs unit test in the SellersRanking class. Attributes ---------- self.ranking: obj Object of the SellersRanking class. self.sellers: list Sellers sales information.""" def setUp(self): """Creates an object of the CloudCost class before each test method.""" self.ranking = SellersRanking() self.sellers = [ {'name': 'Joaquina', 'store': 2, 'value': 1200.00}, {'name': 'Pedro', 'store': 2, 'value': 120.00}, {'name': 'Maria', 'store': 1, 'value': 450.00}, {'name': 'Fernanda', 'store': 1, 'value': 4000.00}, {'name': 'Patricia', 'store': 1, 'value': 100.00}, ] def test_best_seller(self): """Tests if the best seller was returned correctly.""" self.assertEqual(self.ranking.best_seller([]), []) self.assertEqual(self.ranking.best_seller(self.sellers), ['Fernanda']) def test_ranking_list(self): """Tests if the rank list was generated correctly.""" self.assertEqual(len(self.ranking.ranking_list([])), 0) self.assertEqual(len(self.ranking.ranking_list(self.sellers)), 5) def test_best_seller_store(self): """Tests if the best seller was returned correctly according to his store.""" self.assertEqual(len(self.ranking.best_seller_store([], 1)), 0) self.assertEqual(len(self.ranking.best_seller_store(self.sellers, 1)), 1) def test_sales_goals(self): """Tests sellers who didn't reach sales goal.""" self.assertEqual(len(self.ranking.sales_goals([])), 0) self.assertEqual(len(self.ranking.sales_goals(self.sellers)), 3)