class CheckoutTest(unittest.TestCase):
def setUp(self):
self.o = Checkout()
self.item1 = Item(4206, 'Green Onion', 0.89)
self.item2 = Item(4131, 'Fuji Apple', 1.99)
self.item3 = Item(4011, 'Banana', 0.79)
self.purchaseitem1 = PurchaseItem(self.item1, 2)
self.purchaseitem2 = PurchaseItem(self.item2, 4)
self.purchaseitem3 = PurchaseItem(self.item3, 3)
self.o.addPurchaseItems(self.purchaseitem1)
self.o.addPurchaseItems(self.purchaseitem2)
self.o.addPurchaseItems(self.purchaseitem3)
def test_addandgetPurchaseItems(self):
self.oitems = self.o.getPurchaseItems()
self.assertEqual(len(self.oitems), 3)
def test_CalculateTotal(self):
payment = self.o.calcTotal()
self.assertEqual(str(payment), 'Subtotal: 13.89')
