详解Python中Pytest和Unittest的区别

2023-03-13 11:32

短信预约 -IT技能 免费直播动态提醒

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更胜一筹。