ascend-ai-coding Pytest Writer
Overview
使用pytest框架编写高质量、可维护的Python测试用例。支持简单断言、自动发现、fixtures、参数化测试等pytest核心特性。
Quick Start
基本测试结构
# test_sample.py
def inc(x):
return x + 1
def test_answer():
assert inc(3) == 4
运行测试：
pytest
测试文件命名规范
- 测试文件：
test_*.py或*_test.py - 测试函数：
test_* - 测试类：
Test*
Core Capabilities
1. 编写基本测试
使用简单的assert语句，pytest提供详细的失败信息：
def test_addition():
assert 1 + 1 == 2
def test_string_operations():
text = "hello"
assert text.upper() == "HELLO"
assert len(text) == 5
2. 使用Fixtures
Fixtures用于管理测试资源和依赖关系：
import pytest
@pytest.fixture
def sample_data():
return {"name": "test", "value": 42}
def test_with_fixture(sample_data):
assert sample_data["value"] == 42
高级fixtures用法：参见 references/fixtures.md
3. 参数化测试
使用@pytest.mark.parametrize减少重复代码：
@pytest.mark.parametrize("input,expected", [
(3, 4),
(5, 6),
(10, 11),
])
def test_increment(input, expected):
assert input + 1 == expected
更多参数化模式：参见 references/parametrize.md
4. 断言技巧
利用pytest的断言内省功能：
def test_dict_comparison():
expected = {"a": 1, "b": 2}
actual = {"a": 1, "b": 3}
assert actual == expected # pytest会显示详细的差异
断言最佳实践：参见 references/assertions.md
5. 测试组织
按功能组织
tests/
├── test_auth.py
├── test_database.py
├── test_api.py
└── conftest.py # 共享fixtures
使用conftest.py
在conftest.py中定义共享fixtures：
# conftest.py
@pytest.fixture
def db_connection():
conn = create_connection()
yield conn
conn.close()
6. 异常测试
使用pytest.raises测试异常：
def test_zero_division():
with pytest.raises(ZeroDivisionError):
1 / 0
def test_value_error():
with pytest.raises(ValueError) as exc_info:
int("invalid")
assert "invalid literal" in str(exc_info.value)
7. 跳过和预期失败
@pytest.mark.skip(reason="功能未实现")
def test_not_implemented():
pass
@pytest.mark.skipif(sys.version_info < (3, 8), reason="需要Python 3.8+")
def test_python38_feature():
pass
@pytest.mark.xfail
def test_known_failure():
assert False # 预期失败
Testing Best Practices
测试命名
- 使用描述性的测试名称：
test_user_login_with_valid_credentials - 遵循AAA模式：Arrange（准备）→ Act（执行）→ Assert（断言）
def test_calculate_total_with_discount():
# Arrange
cart = Cart(items=[Item(price=100), Item(price=50)])
discount = 0.1
# Act
total = cart.calculate_total(discount)
# Assert
assert total == 135
测试隔离
- 每个测试应该独立运行
- 使用fixtures确保测试间的隔离
- 避免测试间的依赖关系
更多最佳实践：参见 references/best-practices.md
Running Tests
基本命令
# 运行所有测试
pytest
# 运行特定文件
pytest tests/test_auth.py
# 运行特定测试函数
pytest tests/test_auth.py::test_login
# 运行匹配模式的测试
pytest -k "login"
# 显示详细输出
pytest -v
# 显示print语句输出
pytest -s
# 在第一个失败时停止
pytest -x
# 运行上次失败的测试
pytest --lf
测试覆盖率
# 生成覆盖率报告
pytest --cov=src
# 生成HTML报告
pytest --cov=src --cov-report=html
Debugging Failed Tests
查看详细错误信息
pytest -v --tb=long
使用pdb调试
def test_debugging():
import pdb; pdb.set_trace()
result = complex_calculation()
assert result == expected
或在命令行：
pytest --pdb
Resources
references/
详细参考文档：
- fixtures.md - fixtures的完整用法、作用域、参数化fixtures
- parametrize.md - 参数化测试的各种模式和技巧
- assertions.md - 断言技巧、自定义断言消息
- best-practices.md - 测试组织、命名规范、测试隔离等最佳实践
assets/
- test_template.py - 标准测试文件模板，包含常用模式和结构