def test_take_money_when_single_bill_is_added(self): test_cashdesk = CashDesk() test_cashdesk.take_money(Bill(10)) expected_result = 1 self.assertEqual(test_cashdesk.number_of_bills(), expected_result)
def test_take_money_when_a_billbatch_is_added(self): test_cashdesk = CashDesk() test_cashdesk.take_money(BillBatch([Bill(10), Bill(10), Bill(20)])) expected_result = 3 self.assertEqual(test_cashdesk.number_of_bills(), expected_result)