def test_get_currency_amount(print_mock, input_mock): input_mock.side_effect = ['A', '1000'] atm = ATM() result = atm._get_currency_amount() assert result == '1000' print_mock.assert_called_once()