在 python 中进行单元测试

单元测试（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