基于unittest编写单元测试代码

好的，下面是一个简单的例子：

假设我们有一个函数 `add`，实现两个整数相加的功能，我们需要编写一个单元测试来确保它的正确性。

import unittest

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

class TestAdd(unittest.TestCase):
    
    def test_add_positive_numbers(self):
        self.assertEqual(add(2, 3), 5)
    
    def test_add_negative_numbers(self):
        self.assertEqual(add(-2, -3), -5)
    
    def test_add_zero(self):
        self.assertEqual(add(0, 0), 0)

if __name__ == '__main__':
    unittest.main()

这里我们使用了 `unittest` 模块来编写单元测试，它提供了一些方便的断言方法，如 `assertEqual`、`assertTrue`、`assertFalse` 等，可以方便地进行测试。

我们定义了一个 `TestAdd` 类，继承自 `unittest.TestCase`，其中每个以 `test_` 开头的方法都是一个测试用例，用来测试 `add` 函数的不同输入情况下的输出是否正确。

在 `test_add_positive_numbers` 方法中，我们调用 `add(2, 3)`，并使用 `self.assertEqual` 断言其返回值是否等于 5，如果不等于，将会抛出 `AssertionError` 异常，表示测试失败。

最后运行该测试文件，将会执行所有的测试用例，输出测试结果。