要在Python中测试add方法,首先需要编写一个包含该方法的函数或类
- 创建一个名为
calculator.py
的文件,然后添加以下代码:
class Calculator: def add(self, a, b): return a + b
- 创建一个名为
test_calculator.py
的文件,用于编写测试用例。首先,导入unittest
模块和要测试的Calculator
类:
import unittest from calculator import Calculator
- 创建一个继承自
unittest.TestCase
的测试类,并编写针对add
方法的测试用例。示例如下:
class TestCalculator(unittest.TestCase): def setUp(self): self.calculator = Calculator() def test_add_positive_numbers(self): result = self.calculator.add(5, 7) self.assertEqual(result, 12) def test_add_negative_numbers(self): result = self.calculator.add(-3, -6) self.assertEqual(result, -9) def test_add_mixed_numbers(self): result = self.calculator.add(4, -2) self.assertEqual(result, 2)
- 在命令行中运行测试用例:
python -m unittest test_calculator.py
这将运行所有以test_
开头的方法,并报告测试结果。如果所有测试都通过,则会显示OK
。
注意:本示例使用了Python内置的unittest
库。还可以使用其他第三方库(如pytest
)进行测试,但基本原理和步骤相同。