示例#1
0
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')