Python 单元测试：assertTrue 是真值，assertFalse 是假值

在这篇文章中，我们将介绍单元测试的布尔断言方法 assertTrue 和 assertFalse 与身份断言 assertIs 之间的区别。

定义

下面是目前单元测试模块文档中关于 assertTrue 和 assertFalse 的说明，代码进行了高亮：

assertTrue(expr, msg=None)

assertFalse(expr, msg=None)

测试该表达式是真值（或假值）。

注：这等价于

bool(expr) is True

而不等价于

expr is True

（后一种情况请使用 assertIs(expr, True)）。

Mozilla 开发者网络中定义 真值 如下：

在一个布尔值的上下文环境中能变成“真”的值

在 Python 中等价于：

1. bool(expr) is True

这个和 assertTrue 的测试目的完全匹配。

因此该文档中已经指出 assertTrue 返回真值，assertFalse 返回假值。这些断言方法从接受到的值构造出一个布尔值，然后判断它。同样文档中也建议我们根本不应该使用 assertTrue 和 assertFalse。

在实践中怎么理解？

我们使用一个非常简单的例子 - 一个名称为 always_true 的函数，它返回 True。我们为它写一些测试用例，然后改变代码，看看测试用例的表现。

作为开始，我们先写两个测试用例。一个是“宽松的”：使用 assertTrue 来测试真值。另外一个是“严格的”：使用文档中建议的 assertIs 函数。

1. import unittest
2. from func import always_true
4. class TestAlwaysTrue(unittest.TestCase):
5. def test_assertTrue(self):
6. """
7. always_true returns a truthy value
8. """
9. result = always_true()
11. self.assertTrue(result)
13. def test_assertIs(self):
14. """
15. always_true returns True
16. """
17. result = always_true()
19. self.assertIs(result, True)

下面是 func.py 中的非常简单的函数代码：

1. def always_true():
2. """
3. I'm always True.
5. Returns:
6. bool: True
7. """
8. return True

当你运行时，所有测试都通过了：

1. always_true returns True ... ok
2. always_true returns a truthy value ... ok
4. ----------------------------------------------------------------------
5. Ran 2 tests in 0.004s
7. OK

开心ing~

现在，某个人将 always_true 函数改变成下面这样：

1. def always_true():
2. """
3. I'm always True.
5. Returns:
6. bool: True
7. """
8. return 'True'

它现在是用返回字符串 "True" 来替代之前反馈的 True （布尔值）。（当然，那个“某人”并没有更新文档 - 后面我们会增加难度。）

这次结果并不如开心了：

1. always_true returns True ... FAIL
2. always_true returns a truthy value ... ok
4. ======================================================================
5. FAIL: always_true returns True
6. ----------------------------------------------------------------------
7. Traceback (most recent call last):
8. File "/tmp/assertttt/test.py", line 22, in test_is_true
9. self.assertIs(result, True)
10. AssertionError: 'True' is not True
12. ----------------------------------------------------------------------
13. Ran 2 tests in 0.004s
15. FAILED (failures=1)

只有一个测试用例失败了！这意味着 assertTrue 给了我们一个误判false-positive。在它不应该通过测试时，它通过了。很幸运的是我们第二个测试是使用 assertIs 来写的。

因此，跟手册上了解到的信息一样，为了保证 always_true 的功能和更严格测试的结果保持一致，应该使用assertIs 而不是 assertTrue。

使用断言的辅助方法

使用 assertIs 来测试返回 True 和 False 来冗长了。因此，如果你有个项目需要经常检查是否是返回了True 或者 False，那们你可以自己编写一些断言的辅助方法。

这好像并没有节省大量的代码，但是我个人觉得提高了代码的可读性。

1. def assertIsTrue(self, value):
2. self.assertIs(value, True)
4. def assertIsFalse(self, value):
5. self.assertIs(value, False)

总结

一般来说，我的建议是让测试越严格越好。如果你想测试 True 或者 False，听从文档的建议，使用assertIs。除非不得已，否则不要使用 assertTrue 和 assertFalse。

如果你面对的是一个可以返回多种类型的函数，例如，有时候返回布尔值，有时候返回整形，那么考虑重构它。这是代码的异味。在 Python 中，抛出一个异常比使用 False 表示错误更好。

此外，如果你确实想使用断言来判断函数的返回值是否是真，可能还存在第二个代码异味 - 代码是正确封装了吗？如果 assertTrue 和 assertFalse 是根据正确的 if 语句来执行，那么值得检查下你是否把所有你想要的东西都封装在合适的位置。也许这些 if 语句应该封装在测试的函数中。

测试开心！原文发布时间为：2017-10-31

本文来自云栖社区合作伙伴“Linux中国”