回顾一下问题，对于下面这种简单的hello world代码，我们应该怎么测试呢？

print('hello world')

下面是我的一些思路，抛砖引玉。

注意:下面的代码均在python2.7中运行通过，python3可能无法运行

重构代码，测试输出

第一种方式很容易想到，上面的代码没有办法很好的进行测试是因为我们不太好去自动判断输出。

如果能够拿到具体的输出，然后判断输出是否符合预期，这样测试就相对容易了。

这里我们需要先假设print方法是python的内置方法，是可以信任，不需要进行测试的，那么我们的测试点就变成了测试print方法的入参是不是我们所期望的hello world字符串。

重构代码，达到我们的目的，并进行测试

# test_hello_world.py
import unittest

def return_hello_world():
    return 'hello world'

print(return_hello_world())


class HelloWorldTest(unittest.TestCase):

    def test_return_hello_world(self):
        self.assertEqual('hello world', return_hello_world())


if __name__ == '__main__':
    unittest.main()

拿到标准输出里的内容做断言

其实print是将参数中的内容打印到标准输出设备，我们只要知道标准输出设备究竟输出了什么内容就可以比较容易的对简单的hello world进行测试了。

参考这个回答，我们将实现调用print('hello world')，然后拿到标准输出中的内容，再去判断标准输出中的内容是否是hello world，代码如下

# test_hello_world_1.py

import unittest
import sys
from cStringIO import StringIO
# for python 3
# from io import StringIO

class Capturing(list):
    def __enter__(self):
        self._stdout = sys.stdout
        sys.stdout = self._stringio = StringIO()
        return self
    def __exit__(self, *args):
        self.extend(self._stringio.getvalue().splitlines())
        del self._stringio    # free up some memory
        sys.stdout = self._stdout


class TestHelloWorld1(unittest.TestCase):

    def test_hello_world(self):
        with Capturing() as output:
            print("hello world")
            print("hello python")
            print("hello python 2.7")

        self.assertEqual("hello world", output[0])
        self.assertEqual("hello python", output[1])
        self.assertEqual("hello python 2.7", output[-1])

if __name__ == '__main__':
    unittest.main()

使用pytest框架进行测试

pytest框架自带了捕获标准输出内容的特性，可以参考官方文档

这里贴一下官方文档的示例代码,启发一下思路。

def test_myoutput(capsys): # or use "capfd" for fd-level
    print("hello")
    sys.stderr.write("world\n")
    captured = capsys.readouterr()
    assert captured.out == "hello\n"
    assert captured.err == "world\n"
    print("next")
    captured = capsys.readouterr()
    assert captured.out == "next\n"