测试驱动开发:使用unittest框架编写Python单元测试案例
在Python中,我们通常会使用unittest
模块来编写单元测试。以下是一个简单的例子,我们将测试一个计算平方的功能:
import unittest
# 假设我们要测试的函数是这样定义的:
def square(number):
return number ** 2
class TestSquareFunction(unittest.TestCase):
def test_square_positive_number(self):
result = square(5)
self.assertEqual(result, 25)
def test_square_negative_number(self):
result = square(-3)
self.assertEqual(result, 9)
if __name__ == '__main__':
unittest.main()
在这个例子中,我们首先定义了一个square
函数,然后创建了一个TestSquareFunction
测试类,对square
函数进行了测试。
还没有评论,来说两句吧...