单元测试(unit test)就是编写测试来验证某一模块的功能正确性。一般会指定输入,验证输出是否符合预期。
单元测试
进行单元测试,首先要导入 unittest
库。
import unittest
先写一个功能函数,这里以完成加法为例,完成两个数的加法。
def add(a, b):
return a + b
为了验证加法函数的功能是否正确,首先创建一个 TestAdd
类,继承类 unittest.TestCase
,然后在这个类中定义相应的测试函数 test_add()
,测试函数要以 test
开头。在函数内部,通常使用 assertEqual()
、assertTrue()
、assertFalse()
和 assertRaise()
等 assert
断言语句进行验证。
class TestAdd(unittest.TestCase):
def test_add(self):
a = 1
b = 2
self