温馨提示×

如何测试Python中的add方法

小樊
87
2024-08-30 10:16:00
栏目: 编程语言

要在Python中测试add方法,首先需要编写一个包含该方法的函数或类

  1. 创建一个名为calculator.py的文件,然后添加以下代码:
class Calculator:
    def add(self, a, b):
        return a + b
  1. 创建一个名为test_calculator.py的文件,用于编写测试用例。首先,导入unittest模块和要测试的Calculator类:
import unittest
from calculator import Calculator
  1. 创建一个继承自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)
  1. 在命令行中运行测试用例:
python -m unittest test_calculator.py

这将运行所有以test_开头的方法,并报告测试结果。如果所有测试都通过,则会显示OK

注意:本示例使用了Python内置的unittest库。还可以使用其他第三方库(如pytest)进行测试,但基本原理和步骤相同。

0