pytest代码演示
时间: 2024-04-22 16:20:52 浏览: 65
当使用pytest进行代码测试时,可以通过编写测试函数和使用pytest提供的装饰器来定义测试用例。下面是一个简单的pytest代码演示:
```python
# test_calc.py
def add(x, y):
return x + y
def subtract(x, y):
return x - y
def multiply(x, y):
return x * y
def divide(x, y):
if y == 0:
raise ValueError("Cannot divide by zero!")
return x / y
def test_add():
assert add(2, 3) == 5
assert add(5, -2) == 3
def test_subtract():
assert subtract(5, 3) == 2
assert subtract(10, 5) == 5
def test_multiply():
assert multiply(2, 3) == 6
assert multiply(4, -2) == -8
def test_divide():
assert divide(10, 2) == 5
assert divide(8, 4) == 2
def test_divide_by_zero():
try:
divide(10, 0)
except ValueError as e:
assert str(e) == "Cannot divide by zero!"
```
在上面的示例中,我们定义了一些简单的数学运算函数(add、subtract、multiply、divide),并编写了对应的测试函数(test_add、test_subtract、test_multiply、test_divide)。每个测试函数使用了pytest提供的assert语句来断言函数的返回值是否符合预期。
要运行这些测试用例,只需在命令行中执行`pytest`命令,pytest会自动查找并执行以`test_`开头的函数。
阅读全文