要在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
)进行测试,但基本原理和步骤相同。