如何使用输入调用测试函数？

我有一个用Python编写的控制台程序.它使用以下命令询问用户问题：

some_input = input('Answer the question:', ...)

如何使用pytest测试包含输入调用的函数？
我不想强迫测试人员多次输入文本只完成一次测试运行.

解决方法:

您应该模拟内置的input功能,您可以使用pytest提供的teardown功能在每次测试后恢复到原始输入功能.

import module  # The module which contains the call to input

class TestClass:

    def test_function_1(self):
        # Override the Python built-in input method 
        module.input = lambda: 'some_input'
        # Call the function you would like to test (which uses input)
        output = module.function()  
        assert output == 'expected_output'

    def test_function_2(self):
        module.input = lambda: 'some_other_input'
        output = module.function()  
        assert output == 'another_expected_output'        

    def teardown_method(self, method):
        # This method is being called after each test case, and it will revert input back to original function
        module.input = input  

更优雅的解决方案是将mock模块与with statement一起使用.这样您就不需要使用拆卸,并且修补后的方法只能在with范围内使用.

import mock
import module

def test_function():
    with mock.patch.object(__builtins__, 'input', lambda: 'some_input'):
        assert module.function() == 'expected_output'