快捷導(dǎo)航

詳解Python中Pytest和Unittest的區(qū)別

更新時(shí)間：2023年03月13日 11:42:55 作者：田辛?|?田豆芽

Pytest?和?Unittest是Python中屬于最常用的兩個(gè)測試框架。那么他們有些什么區(qū)別呢？?Playwright?為什么只給了Pytest的深度支持，而不是Unittest呢？本文就來和大家詳細(xì)聊聊

1.安裝和使用

在安裝方面， Unittest肯定更好，因?yàn)椴挥冒惭b。 Unittest屬于Python標(biāo)準(zhǔn)庫，裝Python的時(shí)候就安裝了。而Pytest的安裝需要通過pip安裝即可，也不算復(fù)雜。以上是安裝，那么使用呢？

在使用上Pytest更加靈活，可以在命令行中使用各種選項(xiàng)來執(zhí)行測試，而Unittest則需要在腳本中編寫測試用例并使用Unittest模塊來運(yùn)行測試。這一點(diǎn)上Pytest加分較多。

2.編寫測試用例

Pytest和Unittest在編寫測試用例方面也有一些區(qū)別。相對于Unittest而言，Pytest的測試用例編寫更加簡潔。Pytest使用Python的assert關(guān)鍵字來斷言測試結(jié)果，而Unittest則需要使用assertEqual、assertTrue等方法來進(jìn)行斷言。

下面是一個(gè)使用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必須創(chuàng)建測試類，所以絕大多數(shù)場景下，討論代碼簡潔性Pytest要更加的加分。而且我們可以對比一下輸出：

下面是Pytest的輸出結(jié)果

============================= 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的輸出結(jié)果

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

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

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

對比發(fā)現(xiàn)，pytest會(huì)輸出詳細(xì)的結(jié)果，而unittest給出的是整體的判斷。所以，從友好性上來說也是Pytest更勝一籌。