写一个pytest的自动化框架代码

以下是一个简单的pytest自动化测试框架代码示例：

import pytest

class TestCalculator:
    def test_addition(self):
        assert 2 + 2 == 4

    def test_subtraction(self):
        assert 4 - 2 == 2

    def test_multiplication(self):
        assert 2 * 2 == 4

    def test_division(self):
        assert 4 / 2 == 2

if __name__ == "__main__":
    pytest.main()

在这个示例中，我们定义了一个名为TestCalculator的测试类，其中包含了四个测试用例。每个测试用例都是一个独立的函数，使用assert语句来验证预期结果是否等于实际结果。

最后，我们使用pytest.main()函数来运行所有测试用例。pytest会自动发现并执行所有以test_开头的测试用例函数，并输出测试结果。如果有任何测试用例失败，pytest会显示失败的详细信息。