pip install pytest

[pytest]
addopts = -vs
testpaths = cases
python_files = test03.py

import pytest
class TestExample:
    def setup_method(self):
        print("Setup: Before each test")
    def teardown_method(self):
        print("Teardown: After each test")
    def test_example1(self):
        print("Running test_example1")
    def test_example2(self):
        print("Running test_example2")

class TestExample:
    def setup_class(self):
        print("Setup: Before all test")
    def teardown_class(self):
        print("Teardown: After all test")
    def test_example1(self):
        print("Running test_example1")
    def test_example2(self):
        print("Running test_example2")

assert 條件, 錯誤信息

# 斷言整數(shù)
a = 1
b = 2
assert a == b, "a和b不相等"
# 斷言字符串
str = "hello"
assert "hello" == str

def test():
    # 斷言列表
    expect_list = [1, 'apple', 3.14]
    actual_list = [1, 'apple', 3.14]
    # 斷言元組
    expect_tuple = (1, 'apple', 3.14)
    actual_tuple = (1, 'apple', 3.14)
    # 斷言字典
    expect_dict = {'name': 'Alice', 'age': 25}
    actual_dict = {'name': 'Alice', 'age': 25}
    # 斷言集合
    expect_set = {1, 2, 3, 'apple'}
    actual_set = {1, 2, 3, 'apple'}
    assert expect_list == actual_list
    assert expect_tuple == actual_tuple
    assert expect_dict == actual_dict
    assert expect_set == actual_set

def divide(a, b):
    assert b != 0, "除數(shù)不能為0"
    return a / b
# 正常情況
print(divide(10, 2))  # 輸出 5.0
# 觸發(fā)斷言
print(divide(10, 0))  # 拋出 AssertionError: 除數(shù)不能為0

import requests
from jsonschema.validators import validate
# 斷言接口返回值完整字段和值
def test1():
    url = "http://jsonplaceholder.typicode.com/posts/1"
    r = requests.get(url=url)
    expect_data = {
        "userId": 1,
        "id": 1,
        "title": "sunt aut facere repellat provident occaecati excepturi optio reprehenderit",
        "body": "quia et suscipit\nsuscipit recusandae consequuntur expedita et cum\nreprehenderit molestiae ut ut quas totam\nnostrum rerum est autem sunt rem eveniet architecto"
    }
    print(r.json())
    assert r.json() == expect_data
    assert r.json()['userId'] == 1
# 斷言接口返回值重要字段
def test2():
    url = "http://jsonplaceholder.typicode.com/comments?postId=1"
    r = requests.get(url=url)
    print(r.json())
    assert r.json()[1]['id'] == 1
# 斷言接口HTML返回值
def test3():
    url = "http://jsonplaceholder.typicode.com/"
    r = requests.get(url=url)
    assert "Use your own data" in r.text

import pytest
@pytest.mark.parametrize("test_input, expected", [("3+5", 8), ("2+4", 6), ("6*9", 42)])
def test_eval(test_input, expected):
    assert eval(test_input) == expected

import pytest
@pytest.mark.parametrize("n, expected", [(1, 2), (3, 4)])
class TestClass:
    def test_simple_case(self, n, expected):
        assert n + 1 == expected
    def test_weird_simple_case(self, n, expected):
        assert (n * 1) + 1 == expected

import pytest
pytestmark = pytest.mark.parametrize("n, expected", [(1, 2), (3, 4)])
class TestClass:
    def test_simple_case(self, n, expected):
        assert n + 1 == expected
    def test_weird_simple_case(self, n, expected):
        assert (n * 1) + 1 == expected

def data_provider():
    return ["a", "b"]
# 定義測試函數(shù)，依賴自定義數(shù)據(jù)源
@pytest.mark.parametrize("data", data_provider())
def test_data(data):
    assert data is not None
    print(f"Testing with data provider: {data}")