Pytest单元测试框架之简单操作示例
前言:
Pytest是第三方单元格测试框架,更加简单,灵活,而且提供了更多丰富的扩展;
Pytest与UnitTest框架的区别
UnitTest测试用例执行顺序是依照ascii码执行,而Pytest是根据测试用例顺序执行;
1、Pytest官方网站: https://docs.pytest.org/en/latest/
2、Pytest支持 pip 安装,pip3 install pytest,安装成功后直接导入包使用,如: import pytest
3、在Pytest中,它会寻找以test开头或test结尾的测试模块(test**.py、**test.py),然后在模块中执行以test开头的测试方法代码,依据这个进行编写测试用例
4、断言:在UnitTest单元测试框架中提供了丰富的断言方法,如assertEqual() 、assertIn()、assertTrue()、assertIs()等;
在Pytest单元测试框架并没有提供专门的断言方法,而是直接使用Python的assert进行断言
# 导入包
import pytest
#功能:用于计算a与b相加的和
def add(a,b):
return a + b
#功能:用于判断素数
def is_prime(n):
if n < 1:
return False
for i in range(2,n):
if n % i == 0:
return False
return True
#测试相等
def test_add_1():
assert add(3,4) == 7
#测试不相等
def test_add_2():
assert add(17,22) != 50
#测试大于或者等于
def test_add_3():
assert add(17,22) <= 50
#测试小于或者等于
def test_add_4():
assert add(17,22) >= 38
#测试包含
def test_in():
a = 'Hello'
b = 'He'
assert b in a
#测试不包含
def test_not_in():
a = 'Hello'
b = 'hi'
assert b not in a
#判断是否为True
def test_true_1():
assert is_prime(13)
#判断是否为True
def test_true_2():
assert is_prime(7) is True
#判断是否不为True
def test_true_3():
assert is_prime(4) is False
#判断是否不为True
def test_true_4():
assert is_prime(6) is not True
#判断是否为False
def test_false_1():
assert is_prime(8) is False
if __name__ == '__main__':
# main()方法默认执行当前模块中所有以”test开头或test结尾“的函数
pytest.main()
# 若执行终端窗口中使用pytest执行看到执行用例的进度条的话,则需安装 pip3 install pytest-sugar
见执行后输出结果:
其实在一个模块中,不仅包含了函数,还有类,下面来写一段代码实例
# 导入包
import pytest
#功能:用于计算a与b相加的和
def add(a,b):
return a + b
#测试相等
def test_add_1():
assert add(3,4) == 7
#测试不相等
def test_add_2():
assert add(17,22) != 50
class TestAdd(object):
# 测试大于或者等于
def test_add_3(self):
assert add(17, 22) <= 50
# 测试小于或者等于
def test_add_4(self):
assert add(17, 22) >= 38
if __name__ == '__main__':
pytest.main(['-s','-v','test_assert.py'])
见执行后输出结果:
总结:
1、要执行的测试模块必须以test开头
2、要执行的测试函数必须以test开头或test结尾
3、若类要被Pytest执行,那么该类名称首字母必须是:Test,否则不会被执行