编写单元测试

断言语句

1assert 条件, 信息

1def addition(x, y):
2    return x + y

pytest 命令

1pytest

1pytest test.py

1from addition import addition
2
3def test_addition_int():
4    assert addition(4, 5) == 9
5    assert addition(12, 2) == 14
6
7def test_addition_str():
8    assert addition('a', 'b') == 'ab'

1from addition import addition
2
3def test_addition():
4    assert addition(4, 5) == 9
5    assert addition('a', 'b') == 'ab'

1from addition import addition
2
3def test_addition():
4    result = addition(4, 5)
5    expected = 10
6    assert result == expected, f"期望值为 {expected}，但实际得到 {result}"

1from my_class import MyClass
2
3# 方法 1
4class TestMyClass1:
5    def test_addition(self):
6        cls_ = MyClass()
7        result = cls_.add(3, 4)
8        assert result == 7
9
10    def test_subtraction(self):
11        cls_ = MyClass()
12        result = cls_.subtract(7, 3)
13        assert result == 4
14
15# 方法 2
16class TestMyClass2:
17    def setup_method(self):
18        self.cls_ = MyClass()
19    
20    def test_addition(self):
21        result = self.cls_.add(3, 4)
22        assert result == 7
23
24    def test_subtraction(self):
25        result = self.cls_.subtract(7, 3)
26        assert result == 4