如何在Python unittest

时间:2018-03-15 07:31:06

标签: python unit-testing

在我的release_exporter/version.py __version__ = '1.0.6'我想要嘲笑stdoutcheck_version()将最新版本打印到stdout

我做了

模仿stdout
@patch('sys.stdout', new_callable=io.StringIO)
def assert_stdout_1(self, n, expected_output, mock_stdout):
    request = check_version()
    self.assertIn(expected_output, mock_stdout.getvalue())

哪种方法正常。

我尝试做这样的事情来模拟版本号:

def test_check_version_pass(self):
    version.__version__ = '1'

    self.assert_stdout_1('', 'New version')

我知道这是错的,因为我没有嘲笑它(尽管这样可行)。然后我尝试了

@patch.object(version.__version__, 'version.__version__', return_value='1')
def test_check_version_pass(self):

    self.assert_stdout_1('', 'New version')

但我收到AttributeError: 1.0.6 does not have the attribute 'version.__version__'错误。我也尝试使用@patch('release_exporter.version.__version__', return_value='1'),但收到的错误为TypeError: test_check_version_pass() takes 1 positional argument but 2 were given

我不知道我做错了什么。救命啊!

1 个答案:

答案 0 :(得分:0)

好的。我知道了。使用@patch.object()装饰器时,请始终确保您正在使用此方法的参数。例如,在我的情况下(感谢@Sraw):

@patch.object(version, '__version__', return_value='1')
def test_check_version_pass(self, n):

    self.assert_stdout_1('', 'New version')

在这种情况下,我使用了任意参数n

相关问题