def test_receipt_total_price(self):
toothbrush = Product("toothbrush", ProductUnit.EACH)
apples = Product("apples", ProductUnit.KILO)
toothbrush_discount = Discount(toothbrush, "my_description", 0.2)
receipt_toothbrush = ReceiptItem(toothbrush, 2, 1.5, 3)
receipt_apples = ReceiptItem(apples, 3, 1, 3)
receipt = Receipt()
receipt.discounts = [toothbrush_discount]
receipt.items = [receipt_toothbrush, receipt_apples]
totalprice = receipt.total_price()
self.assertEqual([toothbrush_discount], receipt.discounts)
self.assertEqual(5.8, totalprice)
