gpt4 book ai didi

python - 如何在变量中捕获单元测试的标准输出/标准错误?

转载 作者:行者123 更新时间:2023-12-02 02:44:44 27 4
gpt4 key购买 nike

如何在变量中捕获单元测试的标准输出/标准错误?我需要捕获以下单元测试的整个输出输出并将其发送到 SQS。我试过这个:

import unittest, io
from contextlib import redirect_stdout, redirect_stderr


class LogProcessorTests(unittest.TestCase):
def setUp(self):
self.var = 'this value'

def test_var_value(self):
with io.StringIO() as buf, redirect_stderr(buf):
print('Running LogProcessor tests...')
print('Inside test_var_value')
self.assertEqual(self.var, 'that value')
print('-----------------------')
print(buf.getvalue())

但是,它不起作用,并且以下输出仅出现在 stdout/stderr 上。
Testing started at 20:32 ...
/Users/myuser/Documents/virtualenvs/app-venv3/bin/python3 "/Applications/PyCharm CE.app/Contents/helpers/pycharm/_jb_unittest_runner.py" --path /Users/myuser/Documents/projects/application/LogProcessor/tests/test_processor_tests.py
Launching unittests with arguments python -m unittest /Users/myuser/Documents/projects/application/LogProcessor/tests/test_processor_tests.py in /Users/myuser/Documents/projects/application/LogProcessor/tests
Running LogProcessor tests...
Inside test_var_value

that value != this value

Expected :this value
Actual :that value
<Click to see difference>

Traceback (most recent call last):
File "/Applications/PyCharm CE.app/Contents/helpers/pycharm/teamcity/diff_tools.py", line 32, in _patched_equals
old(self, first, second, msg)
File "/usr/local/Cellar/python/3.7.3/Frameworks/Python.framework/Versions/3.7/lib/python3.7/unittest/case.py", line 839, in assertEqual
assertion_func(first, second, msg=msg)
File "/usr/local/Cellar/python/3.7.3/Frameworks/Python.framework/Versions/3.7/lib/python3.7/unittest/case.py", line 1220, in assertMultiLineEqual
self.fail(self._formatMessage(msg, standardMsg))
File "/usr/local/Cellar/python/3.7.3/Frameworks/Python.framework/Versions/3.7/lib/python3.7/unittest/case.py", line 680, in fail
raise self.failureException(msg)
AssertionError: 'this value' != 'that value'
- this value
? ^^
+ that value
? ^^

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/usr/local/Cellar/python/3.7.3/Frameworks/Python.framework/Versions/3.7/lib/python3.7/unittest/case.py", line 59, in testPartExecutor
yield
File "/usr/local/Cellar/python/3.7.3/Frameworks/Python.framework/Versions/3.7/lib/python3.7/unittest/case.py", line 615, in run
testMethod()
File "/Users/myuser/Documents/projects/application/LogProcessor/tests/test_processor_tests.py", line 15, in test_var_value
self.assertEqual(self.var, 'that value')



Ran 1 test in 0.004s

FAILED (failures=1)

Process finished with exit code 1

Assertion failed

Assertion failed

任何的想法?如果需要更多信息,请告诉我。

最佳答案

如果您手动实例化测试运行程序(例如 unittest.TextTestRunner ),您可以指定它写入的(文件)流。默认为 sys.stderr ,但您可以使用 StringIO 代替。这将捕获单元测试本身的输出。您自己的打印语句的输出不会被捕获,但您可以使用 redirect_stdout上下文管理器,使用相同的 StringIO 对象。

请注意,我建议避免使用打印语句,因为它们会干扰 unittest 框架的输出(您的测试输出会破坏 unittest 框架的输出行)并且重定向 stdout/stderr 有点麻烦流。更好的解决方案是使用 logging取而代之的是模块。然后,您可以添加一个日志处理程序,将所有日志消息写入 StringIO 以进行进一步处理(在您的情况下:发送到 SQS)。

下面是基于您使用打印语句的代码的示例代码。

#!/usr/bin/env python3

import contextlib
import io
import unittest


class LogProcessorTests(unittest.TestCase):

def setUp(self):
self.var = 'this value'

def test_var_value(self):
print('Running LogProcessor tests...')
print('Inside test_var_value')
self.assertEqual(self.var, 'that value')
print('-----------------------')


if __name__ == '__main__':
# find all tests in this module
import __main__
suite = unittest.TestLoader().loadTestsFromModule(__main__)
with io.StringIO() as buf:
# run the tests
with contextlib.redirect_stdout(buf):
unittest.TextTestRunner(stream=buf).run(suite)
# process (in this case: print) the results
print('*** CAPTURED TEXT***:\n%s' % buf.getvalue())

这打印:
*** CAPTURED TEXT***:
Running LogProcessor tests...
Inside test_var_value
F
======================================================================
FAIL: test_var_value (__main__.LogProcessorTests)
----------------------------------------------------------------------
Traceback (most recent call last):
File "test.py", line 16, in test_var_value
self.assertEqual(self.var, 'that value')
AssertionError: 'this value' != 'that value'
- this value
? ^^
+ that value
? ^^


----------------------------------------------------------------------
Ran 1 test in 0.000s

FAILED (failures=1)

这确认了所有输出(来自 unittest 框架和测试用例本身)都在 StringIO 对象中捕获。

关于python - 如何在变量中捕获单元测试的标准输出/标准错误?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56045623/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com