欢迎访问宙启技术站
智能推送

Python中如何编写函数测试用例(FunctionTestCase())

发布时间:2023-12-26 05:44:57

在Python中,可以使用unittest模块中的FunctionTestCase类来编写函数的测试用例。FunctionTestCase类是TestCase类的子类,它允许我们测试独立的函数而不需要创建测试类。

下面是一个简单的示例,展示如何使用FunctionTestCase编写函数的测试用例:

import unittest

def add_numbers(a, b):
    return a + b

def multiply_numbers(a, b):
    return a * b

def divide_numbers(a, b):
    if b == 0:
        raise ZeroDivisionError("Cannot divide by zero.")
    return a / b

def is_even_number(n):
    return n % 2 == 0

def is_prime_number(n):
    if n < 2:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True

class FunctionTestCase(unittest.TestCase):

    def test_add_numbers(self):
        self.assertEqual(add_numbers(2, 3), 5)
        self.assertEqual(add_numbers(-1, 1), 0)
        self.assertEqual(add_numbers(0, 0), 0)

    def test_multiply_numbers(self):
        self.assertEqual(multiply_numbers(2, 3), 6)
        self.assertEqual(multiply_numbers(-1, 1), -1)
        self.assertEqual(multiply_numbers(0, 0), 0)

    def test_divide_numbers(self):
        self.assertEqual(divide_numbers(6, 3), 2)
        self.assertEqual(divide_numbers(10, 2), 5)
        self.assertRaises(ZeroDivisionError, divide_numbers, 10, 0)

    def test_is_even_number(self):
        self.assertTrue(is_even_number(2))
        self.assertFalse(is_even_number(3))
        self.assertTrue(is_even_number(0))

    def test_is_prime_number(self):
        self.assertTrue(is_prime_number(2))
        self.assertTrue(is_prime_number(3))
        self.assertFalse(is_prime_number(4))
        self.assertTrue(is_prime_number(17))
        self.assertFalse(is_prime_number(1))

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(FunctionTestCase)
    unittest.TextTestRunner(verbosity=2).run(suite)

在这个例子中,我们定义了五个独立函数,并使用FunctionTestCase编写了对应的测试用例。每个测试用例都是一个独立的测试方法,我们可以在其中使用assertEqualassertTrueassertFalse等断言方法来验证函数的行为是否符合预期。

test_divide_numbers测试用例中,我们还使用了assertRaises方法来验证函数在特定条件下是否会产生异常。在这个例子中,我们希望当函数的除数为0时,它会抛出一个ZeroDivisionError异常。

最后,我们使用unittest.TestLoader来加载FunctionTestCase类,并使用unittest.TextTestRunner运行测试用例。通过设置verbosity参数为2,我们可以获得更详细的测试结果输出。

总结起来,使用FunctionTestCase编写函数的测试用例非常简单。我们只需要在测试类中定义对应的测试方法,并使用断言方法来验证函数的行为是否符合预期。在运行测试时,我们可以通过加载测试类并运行测试用例来获取详细的测试结果输出。