Python 之真假“美猴王”

導讀	單元測試的重要性就不多說了，可惡的是Python中有太多的單元測試框架和工具，什麼unittest, testtools, subunit, coverage, testrepository, nose, mox, mock, fixtures, discover，再加上setuptools, distutils等等這些。在這篇文章中，我們將介紹單元測試的布林斷言方法 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 中等價於：

bool(expr) is True

這個和 assertTrue 的測試目的完全匹配。

因此該文件中已經指出 assertTrue 返回真值，assertFalse 返回假值。這些斷言方法從接受到的值構造出一個布林值，然後判斷它。同樣文件中也建議我們根本不應該使用 assertTrue 和 assertFalse。

我們使用一個非常簡單的例子 - 一個名稱為 always_true 的函式，它返回 True。我們為它寫一些測試用例，然後改變程式碼，看看測試用例的表現。

作為開始，我們先寫兩個測試用例。一個是“寬鬆的”：使用 assertTrue 來測試真值。另外一個是“嚴格的”：使用文件中建議的 assertIs 函式。

import unittest
from func import always_true
class TestAlwaysTrue(unittest.TestCase):
    def test_assertTrue(self):
        """
        always_true returns a truthy value
        """
        result = always_true()
        self.assertTrue(result)
    def test_assertIs(self):
        """
        always_true returns True
        """
        result = always_true()
        self.assertIs(result, True)

下面是 func.py 中的非常簡單的函式程式碼：

def always_true():
    """
    I'm always True.
    Returns:
        bool: True
    """
    return True

當你執行時，所有測試都透過了：

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

現在，某個人將 always_true 函式改變成下面這樣：

def always_true():
    """
    I'm always True.
    Returns:
        bool: True
    """
    return 'True'

它現在是用返回字串 "True" 來替代之前反饋的 True （布林值）。（當然，那個“某人”並沒有更新文件 - 後面我們會增加難度。）

這次結果並不如開心了：

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

只有一個測試用例失敗了！這意味著 assertTrue 給了我們一個誤判。在它不應該透過測試時，它透過了。很幸運的是我們第二個測試是使用 assertIs 來寫的。

因此，跟手冊上了解到的資訊一樣，為了保證 always_true 的功能和更嚴格測試的結果保持一致，應該使用 assertIs 而不是 assertTrue。

使用 assertIs 來測試返回 True 和 False 來冗長了。因此，如果你有個專案需要經常檢查是否是返回了 True 或者 False，那們你可以自己編寫一些斷言的輔助方法。

這好像並沒有節省大量的程式碼，但是我個人覺得提高了程式碼的可讀性。

def assertIsTrue(self, value):
    self.assertIs(value, True)
def assertIsFalse(self, value):
    self.assertIs(value, False)

一般來說，我的建議是讓測試越嚴格越好。如果你想測試 True 或者 False，聽從文件的建議，使用 assertIs。除非不得已，否則不要使用 assertTrue 和 assertFalse。

如果你面對的是一個可以返回多種型別的函式，例如，有時候返回布林值，有時候返回整形，那麼考慮重構它。這是程式碼的異味。在 Python 中，丟擲一個異常比使用 False 表示錯誤更好。

此外，如果你確實想使用斷言來判斷函式的返回值是否是真，可能還存在第二個程式碼異味 - 程式碼是正確封裝了嗎？如果 assertTrue 和 assertFalse 是根據正確的 if 語句來執行，那麼值得檢查下你是否把所有你想要的東西都封裝在合適的位置。也許這些 if 語句應該封裝在測試的函式中。

原文來自：