如何在Python中用nosetest / unittest声明输出?

我正在写下一个函数的testing:

def foo(): print 'hello world!' 

所以当我想testing这个函数时,代码将会是这样的:

 import sys from foomodule import foo def test_foo(): foo() output = sys.stdout.getline().strip() # because stdout is an StringIO instance assert output == 'hello world!' 

但是,如果我用-s参数运行noseteststesting崩溃。 如何用unittest或nose模块捕捉输出?

9 Solutions collect form web for “如何在Python中用nosetest / unittest声明输出?”

我使用这个上下文pipe理器来捕获输出。 它通过临时replacesys.stdout最终使用与其他一些答案相同的技术。 我更喜欢上下文pipe理器,因为它把所有的簿记包装成一个单独的函数,所以我不必重写任何try-finally代码,我不必为此编写setup和teardown函数。

 from contextlib import contextmanager from StringIO import StringIO @contextmanager def captured_output(): new_out, new_err = StringIO(), StringIO() old_out, old_err = sys.stdout, sys.stderr try: sys.stdout, sys.stderr = new_out, new_err yield sys.stdout, sys.stderr finally: sys.stdout, sys.stderr = old_out, old_err 

像这样使用它:

 with captured_output() as (out, err): foo() # This can go inside or outside the `with` block output = out.getvalue().strip() self.assertEqual(output, 'hello world!') 

另外,由于原来的输出状态是在退出with块的时候恢复的,所以我们可以在第一个函数中设置第二个捕获块,这是使用setup和teardown函数不可能的,在try-最后手动阻止。 当testing的目标是比较两个函数的相对结果而不是一些预先计算的值时,这种能力就派上用场了。

如果你真的想这样做,你可以在testing期间重新分配sys.stdout。

 def test_foo(): import sys from foomodule import foo from StringIO import StringIO saved_stdout = sys.stdout try: out = StringIO() sys.stdout = out foo() output = out.getvalue().strip() assert output == 'hello world!' finally: sys.stdout = saved_stdout 

但是,如果我正在编写此代码,我宁愿将一个可选的outparameter passing给foo函数。

 def foo(out=sys.stdout): out.write("hello, world!") 

那么testing就简单多了:

 def test_foo(): from foomodule import foo from StringIO import StringIO out = StringIO() foo(out=out) output = out.getvalue().strip() assert output == 'hello world!' 

从2.7版开始,您不再需要重新分配sys.stdout ,这是通过buffer标志提供的。 而且,这是nosetest的默认行为。

这是一个失败的非缓冲上下文示例:

 import sys import unittest def foo(): print 'hello world!' class Case(unittest.TestCase): def test_foo(self): foo() if not hasattr(sys.stdout, "getvalue"): self.fail("need to run in buffered mode") output = sys.stdout.getvalue().strip() # because stdout is an StringIO instance self.assertEquals(output,'hello world!') 

您可以通过--buffer命令行标志-b ,– --bufferunittest.main选项来设置缓冲区。 相反的是通过--nocapture旗帜 – --nocapture来实现的。

 if __name__=="__main__": assert not hasattr(sys.stdout, "getvalue") unittest.main(module=__name__, buffer=True, exit=False) #. #---------------------------------------------------------------------- #Ran 1 test in 0.000s # #OK assert not hasattr(sys.stdout, "getvalue") unittest.main(module=__name__, buffer=False) #hello world! #F #====================================================================== #FAIL: test_foo (__main__.Case) #---------------------------------------------------------------------- #Traceback (most recent call last): # File "test_stdout.py", line 15, in test_foo # self.fail("need to run in buffered mode") #AssertionError: need to run in buffered mode # #---------------------------------------------------------------------- #Ran 1 test in 0.002s # #FAILED (failures=1) 

我只是刚刚学习Python,发现自己正在努力解决与上面类似的问题,unit testing的输出方法。 我上面的foo模块的unit testing已经结束了,看起来像这样:

 import sys import unittest from foo import foo from StringIO import StringIO class FooTest (unittest.TestCase): def setUp(self): self.held, sys.stdout = sys.stdout, StringIO() def test_foo(self): foo() self.assertEqual(sys.stdout.getvalue(),'hello world!\n') 

很多这些答案都失败了,因为你不能from StringIO import StringIO Python 3中的from StringIO import StringIO中获取这个。下面是一个基于@ naxa评论和Python Cookbook的最小工作片段。

 from io import StringIO from unittest.mock import patch with patch('sys.stdout', new=StringIO()) as fakeOutput: print('hello world') self.assertEqual(fakeOutput.getvalue().strip(), 'hello world') 

编写testing通常会向我们展示编写代码的更好方法。 与Shane的回答类似,我想提出另一种看待这个问题的方法。 你真的想断言你的程序输出了一个特定的string,或只是它构造了一个特定的string输出? 这变得更容易testing,因为我们可以假定Python print语句正确地工作。

 def foo_msg(): return 'hello world' def foo(): print foo_msg() 

那么你的testing很简单:

 def test_foo_msg(): assert 'hello world' == foo_msg() 

当然,如果你真的需要testing你的程序的实际输出,那么可以随意忽略。 🙂

在python 3.5中,你可以使用StringIO()StringIO() 。 这是对你的代码的修改

 import contextlib from io import StringIO from foomodule import foo def test_foo(): temp_stdout = StringIO() with contextlib.redirect_stdout(temp_stdout): foo() output = temp_stdout.getvalue().strip() assert output == 'hello world!' 

或者考虑使用pytest,它已经内置了支持stdout和stderr的支持。 看文档

根据Rob Kennedy的回答,我编写了一个基于类的上下文pipe理器来缓冲输出。

用法如下:

 with OutputBuffer() as bf: print('hello world') assert bf.out == 'hello world\n' 

这是实现:

 from io import StringIO import sys class OutputBuffer(object): def __init__(self): self.stdout = StringIO() self.stderr = StringIO() def __enter__(self): self.original_stdout, self.original_stderr = sys.stdout, sys.stderr sys.stdout, sys.stderr = self.stdout, self.stderr return self def __exit__(self, exception_type, exception, traceback): sys.stdout, sys.stderr = self.original_stdout, self.original_stderr @property def out(self): return self.stdout.getvalue() @property def err(self): return self.stderr.getvalue() 
  • 在持续集成构build中无头无虑地运行JavaScriptunit testing
  • 用Mockito嘲笑静态方法
  • Google可以使用智能指针返回types吗?
  • 使用entity framework进行unit testing
  • 我如何编写一个unit testing来确定一个对象是否可以被垃圾收集?
  • unit testing项目可以加载目标应用程序的app.config文件吗?
  • unit testingmultithreading应用程序?
  • 你最喜欢的Delphi嘲笑库是什么?
  • 首选的Pythonunit testing框架
  • 如何用模拟来嘲笑只读属性?
  • 在unit testing中有多个断言是不好的做法?