class TestVendingMachine(unittest.TestCase): def setUp(self): self.vending = VendingMachine(156700) self.stock = {'root beer': 10, 'coca-cola': 10, 'sprite': 30} self.sorted_stock = sorted(self.stock.items(), key=lambda x: x[1]) def get_stock_test (self): self.assertEqual(self.vending.get_stock(),{'coca-cola': 10, 'sprite': 10,'root beer': 10}) def test_replenish_test(self): self.vending.replenish('sprite',20) self.assertEqual(self.sorted_stock,[('coca-cola', 10), ('root beer', 10), ('sprite', 30)]) def test_sell(self): self.vending.sell('sprite') self.assertEqual(self.sorted_stock,[('coca-cola', 10), ('root beer', 10), ('sprite', 30)])