python篇-pytest框架详解

import pytest
 
# 简单的测试函数
def func(x):
    return x + 1
 
# 使用pytest的测试用例
def test_function():
    assert func(3) == 5
 
# 使用pytest.mark.parametrize进行参数化测试
@pytest.mark.parametrize("test_input,expected", [("3", 5), ("5", 6)])
def test_function_parametrized(test_input, expected):
    assert func(int(test_input)) == expected
 
# 使用pytest.raises验证异常
def test_raise():
     with pytest.raises(ValueError):
         func("a")
 
# 运行测试
if __name__ == '__main__':
    pytest.main(['-s', 'test_pytest.py'])

这段代码定义了一个简单的函数func，以及使用pytest编写的几个测试用例。第一个测试用例test_function验证了func的正常行为。第二个测试用例test_function_parametrized使用了pytest的参数化功能来测试多个输入。最后一个测试用例test_raise验证了当传入错误类型时func是否引发异常。最后，代码中通过pytest.main运行了测试用例。