詳解Python中Pytest和Unittest的區別

2023-11-03 10:00:32

1.安裝和使用

在安裝方面， Unittest肯定更好，因為不用安裝。 Unittest屬於Python標準庫，裝Python的時候就安裝了。而Pytest的安裝需要通過pip安裝即可，也不算複雜。以上是安裝，那麼使用呢？

在使用上Pytest更加靈活，可以在命令列中使用各種選項來執行測試，而Unittest則需要在指令碼中編寫測試用例並使用Unittest模組來執行測試。這一點上Pytest加分較多。

2.編寫測試用例

Pytest和Unittest在編寫測試用例方面也有一些區別。相對於Unittest而言，Pytest的測試用例編寫更加簡潔。Pytest使用Python的assert關鍵字來斷言測試結果，而Unittest則需要使用assertEqual、assertTrue等方法來進行斷言。

下面是一個使用Pytest編寫的測試用例範例，測試用例的目的是測試下面的類：

class Calculator:  
    def add(self, a, b):  
        return a + b  
  
    def subtract(self, a, b):  
        return a - b

如果用Pytest編寫

import pytest  
  
from Calc import Calculator  
  
  
@pytest.mark.parametrize("a, b, expected", [  
    (2, 3, 5),  
    (0, 0, 0),  
    (-1, 1, 0),  
])  
def test_calculator_add(a, b, expected):  
    calculator = Calculator()  
    assert calculator.add(a, b) == expected

而使用Unittest編寫的相同測試用例則為：

import unittest  
  
from Calc import Calculator  
  
  
class TestCalculator(unittest.TestCase):  
    def test_calculator_add(self):  
        calculator = Calculator()  
        self.assertEqual(calculator.add(2, 3), 5)  
        self.assertEqual(calculator.add(0, 0), 0)  
        self.assertEqual(calculator.add(-1, 1), 0)

Unittest必須建立測試類，所以絕大多數場景下，討論程式碼簡潔性Pytest要更加的加分。而且我們可以對比一下輸出：

下面是Pytest的輸出結果

============================= test session starts =============================
collecting ... collected 3 items

test_calc.py::test_calculator_add[2-3-5] PASSED [ 33%]
test_calc.py::test_calculator_add[0-0-0] PASSED [ 66%]
test_calc.py::test_calculator_add[-1-1-0] PASSED [100%]

============================== 3 passed in 0.01s ==============================

下面是Unittest的輸出結果

============================= test session starts =============================
collecting ... collected 1 item

u.py::TestCalculator::test_calculator_add PASSED [100%]

============================== 1 passed in 0.01s ==============================

對比發現，pytest會輸出詳細的結果，而unittest給出的是整體的判斷。所以，從友好性上來說也是Pytest更勝一籌。