def test_add_account(input_mock, print_mock): input_mock.side_effect = ['3', '1', '1000.00'] atm = ATM() account = atm.add_account() assert print_mock.call_count > 2 assert account.amount == Decimal('1000.00')