从另一个类获取方法的输出,并在unittest中进行测试

一个简单的解决方案是 remap stdout to a file 并在单元测试类中处理方法执行后的文件。

import sys
sys.stdout = open('result', 'w')

test_code()
# read 'result'

或者,您可以使用 StringIO 单元

import StringIO
output = StringIO.StringIO()
sys.stdout = output

例子:

#!remap.py
import sys
import StringIO

backup_sys = sys.stdout # backup our standard out
output = StringIO.StringIO() # creates file stream for monitoring test result
sys.stdout = output
print 'test' # prints to our IO stream

sys.stdout = backup_sys # remap back to console
print output.getvalue() # prints the entire contents of the IO stream

输出

test

More details on the module can be found here.