文章目录
- 01-单元测试基础
- 什么是单元测试
- 常用的文件结构
- 运行单元测试
- 02. 断言函数
- 03. Test Fixtures
- 什么是Test Fixtures
- 模块级别的Fixtures
- 类级别的Fixtures
- 方法级别的Fixtures
- 04.Mock
01-单元测试基础
- 什么是单元测试
- 常用的文件结构
- 编写第一个单元测试
- 运行单元测试
什么是单元测试
单元测试是指一个自动化的测试:
- 用来验证一小段代码(单元)的正确性,例如只测试某个函数写的是否正确
- 可以快速执行
- 在独立的环境中执行
常用的文件结构
myprj
是项目代码目录tests
是单元测试代码目录,一般来说测试代码与项目代码文件目录结构一样- 在
tests/basic
中test_calculator.py
必须以test_
开头,因为测试的工具会寻找所有以test
开头的文件,把他们当作单元测试文件运行,这算是一个规范
运行单元测试
myprj.basic中calculator.py代码:
class Calculator:def add(self, *args):res = 0for n in args:res += nreturn res
tests.basic中test_calculator.py代码:
import unittest
from myprj.basic.calculator import Calculator
"""文件要用"test_"开头测试类名要用"Test"开头
"""
class TestCalculator(unittest.TestCase):def test_add(self):"""测试类方法必须用"test_"开头通常分为三部分:setupactionassert"""# setupcal = Calculator()excepted_result = 10 # 期望输出为10# actionactual_result = cal.add(2,3,5)# assert 判断执行结果与期望结果是否相等,如果不相等,则测试失败# 说明代码有问题self.assertEqual(excepted_result, actual_result)
为了方便的运行测试:
pip install nose
pip install coverage# 运行一个测试文件
python -m unittest -v tests.basic.test_calculator
# 运行所有测试文件
nosetests --with-doctest -v tests/basic/
# 统计测试覆盖率
nosetests --with-coverage --cover-erase -v tests/
也可以使用pytest
:
pytest --doctest-modules tests/
02. 断言函数
常用断言函数:
03. Test Fixtures
- 什么是Test Fixtures
- 模块级别的Fixtures
- 类级别的Fixtures
- 方法级别的Fixtures
什么是Test Fixtures
在测试之前或者之后执行的函数或者方法被称之为Test Fixtures
比如:在测试前需要先连接数据库,在测试后需要断开与数据建库的连接
模块级别的Fixtures
setUpModule() 在整个模块执行之前需要执行的函数
tearDownModule() 在整个模块执行结束之后需要执行的函数
import unittest
from myprj.fixtures.bank_account import BankAccountdef setUpModule():print("calling setUpModule")def tearDownModule():print("calling tearDownModule")class TestBankAccount(unittest.TestCase):def test_deposit_success(self):bank_account = BankAccount(0)bank_account.deposit(10)self.assertEqual(10, bank_account.balance)def test_withdraw_success(self):bank_account = BankAccount(10)bank_account.withdraw(10)self.assertEqual(0, bank_account.balance)
上述代码中,当单独执行test_deposit_success
与test_withdraw_success
时候,代码会自动执行setUpModule
与 tearDownModule
.
当执行TestBankAccount
进行测试时,虽然有两个测试用例,但是setUpModule
与 tearDownModule
只会执行一次
类级别的Fixtures
setUpClass() 在整类执行之前需要执行的函数
tearDownClass() 在整个类执行结束之后需要执行的函数
import unittest
from myprj.fixtures.bank_account import BankAccountdef setUpModule():print("calling setUpModule")def tearDownModule():print("calling tearDownModule")class TestBankAccount(unittest.TestCase):@classmethoddef setUpClass(cls) -> None:print("calling setUpClass")@classmethoddef tearDownClass(cls) -> None:print("calling tearDownClass")def test_deposit_success(self):bank_account = BankAccount(0)bank_account.deposit(10)self.assertEqual(10, bank_account.balance)def test_withdraw_success(self):bank_account = BankAccount(10)bank_account.withdraw(10)self.assertEqual(0, bank_account.balance)
方法级别的Fixtures
setUp() 在任何一个方法执行之前都会被执行
tearDown()
import unittest
from myprj.fixtures.bank_account import BankAccountclass TestBankAccount(unittest.TestCase):def setUp(self) -> None:'''每一个类方法都要新建一个对象在任何一个方法执行之前都会执行'''print("calling setUp")self.bank_account = BankAccount(10)def tearDown(self) -> None:print("calling tearDown")self.bank_account = Nonedef test_deposit_success(self):self.bank_account.deposit(10)self.assertEqual(20, self.bank_account.balance)def test_withdraw_success(self):self.bank_account.withdraw(10)self.assertEqual(0, self.bank_account.balance)
04.Mock
- 什么是Mock
- Mock和MagicMock
- Mock实例
本文参考:
https://www.bilibili.com/video/BV1SP4y1D7Wd/?spm_id_from=333.999.0.0