会员服务 ·

PyTest入门

2018 年 8 月 17 日 Python程序员

Pytest是我首选的Python测试库。它使简单的测试非常容易编写，它拥有许多高级的功能（和众多插件），有助于更高级的测试方案。

为了演示基本功能，我打算使用py.test演示在测试驱动风格下解决cryptopals challenges的第一集合的第一题。

注意︰

我要去做的第一个的挑战和也许涉及到第二个。如果你想自己测试，可以在阅读这篇文章的其余部分之前这样做。

安装和第一个测试

安装︰

注意

我使用 Python 3.5，并且所有的操作在virtualenv。如果你没有安装Python，或不知道如何使用 pip 和virtualenv ，看看Python部落影音学堂中＜０基础１小时入门Python＞的第一课。

第一个挑战要求我们把一个十六进制编码的字符串转化为Base64编码。我先编写测试，代表所面临的挑战。 py.test默认检查以test_*.py命名的文件名，在文件内部查找以test_打头的方法或函数，并执行它们。所以我创建文件test_set1.py︰

我没弄错吧？

是的，我并没有写任何代码︰第一个挑战超级简单，Python可以使用bytes.fromhex将十六进制编码的字符串解析成bytestring，并且base64模块中有base64 encoding编码。

然而，这只是演示了Pytest的简单部分，测试的只是以test_打头的函数或方法，仅仅使用了简单的assert语句，而不是一系列断言方法（assertEqual, assertNotEqual, assertAlmostEqual）。

接下来是更真实的测试情况

来看一个更完整的示例，第二个挑战是要实现一个函数，异或两个固定长度的缓冲区。我会先写以下测试用例︰

运行测试：

如所料，出现错误——测试程序试图导入cryptopals模块，然而我并未创建这个模块。这个错误看起来不同于失败的测试，因为这发生在pytest调用”collection”阶段，pytest遍历你的文件，寻找测试模块（命名为test_whatever.py的文件）和测试方法（def test_whatever()）。更多关于pytest如何发现测试用例（以及如何自定义测试），请参见

conventions for Python test discovery。

现在我主要来演示什么是失败的测试。在cryptopals.py中：

不出所料，失败了：

（我使用-q即—quiet，简化输出。）

我喜欢这种高亮显示测试失败行的方式，并且显示了断言语句运行的值。

一旦我写了正确的代码（在这里忽略继续挑战的精神吧），应该会看到以下输出：

尝试更高级的pytest：参数化测试函数

对于最后一个例子，来看一下pytest稍微复杂的应用。我想写更多的测试来检查我的fixed_xor函数作用于不同长度的bytestrings。这个挑战只是说此函数应该使用两个相同长度的缓冲区，，但并没有指定当两个缓冲区长度不同时会发生什么。所以，我决定这种情况下结果应该是长度最短的bytestring（主要是因为这使函数写起来简单些）。

为了正确的验证，我测试以下几种不同场景：bs1长度小于 bs2即len(bs2)<len(bs1)，还有bs1或bs2为空的情况，——这种极端的测试用例往往都是bug潜伏的地方。我可以写四个更多的测试函数，但这就会重复，所以我会改为使用参数化的测试函数：