包括用户输入的单元测试功能，while loops且输入是另一个功能输出-Python

Question

我已经为ATM机制作了一个程序，该机器检查用户输入的PIN，允许用户输入数量，如果PIN匹配，然后从Account_balance中提取资金。但是，每个函数似乎取决于：一个函数的返回是另一个函数的参数。这是分别测试每个函数的混乱，当我尝试这样做时，整个程序都会运行并索取每个函数的输入。 由于我的get_pin和get_amount函数在带有用户输入的循环时也具有，因此我觉得这为单位测试所需的方式增加了另一个复杂性。

有办法这样做吗？ 还是您认为最好重构我的代码以删除用户输入和循环，并且每个单独的函数中只有参数？

我要检查的测试将是：

• fort_cash函数 - 检查正确的金额是否已撤回
• get_pin函数 - 检查PIN匹配GET_AMOUNT
• 功能是否匹配Get_Amount函数 - 添加了错误的金额，并带来错误消息。

我的ATM程序是：

user = {
    'pin': 1234
}

def withdraw_cash(amount):
    balance_account = 100
    if amount > balance_account:
        raise ValueError("You don't have sufficient balance to make this withdrawal")
    else:
        new_balance = balance_account - amount
        return new_balance
    
    
def get_pin():
    count = 0
    to_exit = False
    while (count < 3) and (not to_exit):
        try:
            pin = int(input('Please enter your four digit pin: '))
        except ValueError:
            print("Please enter correct pin")
            count += 1
        if pin != user['pin']:
            print("Pin does not match.. Try Again")
            count += 1
        else:
            return get_amount(pin)        
    if count == 3:
        a = '3 UNSUCCESFUL PIN ATTEMPTS, EXITING \n !!!!!YOUR CARD HAS BEEN LOCKED!!!!!'
        return a 
            


def get_amount(pin):
    while True:
        try:
            amount = int(input("Enter the amount of money you want to withdraw: "))
        except ValueError as v:
            print(f"Enter correct amount: ")
        else:
            return withdraw_cash(amount)

try:
    get_pin()
except ValueError as v:
    print(f"ERROR: {v}")