python - Python unittest 一个没有参数的简单用户输入函数

Question

您将如何在 Python 中对以下内容进行单元测试？

• 我只想单元测试输入是一个字符串而不是整数、浮点数或字符。
• self.assertRaises 需要函数的参数，但没有参数，因为函数只要求输入。

def name():
    account_name = input("Please enter your business name: ")
    print('\n'*2)
    return account_name

def nextstep(accts):
    result = input("Would you like to make a deposit or withdrawal?: ").lower()
    if result.startswith("d"):
        amount = int(input("Enter an amount to deposit: "))
        accts.deposit(amount)
    if result.startswith("w"):
        amount = int(input("Enter an amount to withdrawal: "))
        accts.withdraw(amount)
    return accts

def continue_step():
    print('\n')
    answer = input("Would you like to continue?: ").lower()
    if answer.startswith("y"):
        print("Yes")
    else:
        print("Bye")
    return answer

import unittest

class TestSample(unittest.TestCase):

    def test_accountname(self):
        self.assertRaises(TypeError, continue_step, parameter)
    
    unittest.main()

标签： pythonunit-testingpython-unittestpython-unittest.mock