在Ubuntu系统中,Python单元测试通常使用unittest
库来编写。unittest
是Python的标准库之一,提供了一个框架来编写和运行测试用例。
以下是一个简单的Python单元测试示例:
test_sample.py
的文件,用于编写测试用例:import unittest
# 这是你需要测试的模块
def add(a, b):
return a + b
# 创建一个测试类,继承unittest.TestCase
class TestAddFunction(unittest.TestCase):
# 在每个测试方法之前执行
def setUp(self):
print("Setting up the test environment")
# 在每个测试方法之后执行
def tearDown(self):
print("Tearing down the test environment")
# 测试add函数是否正确地将两个数字相加
def test_addition(self):
self.assertEqual(add(1, 2), 3)
self.assertEqual(add(-1, 1), 0)
self.assertEqual(add(0, 0), 0)
# 测试add函数是否正确地处理浮点数
def test_addition_with_floats(self):
self.assertAlmostEqual(add(1.1, 2.2), 3.3, places=1)
if __name__ == '__main__':
unittest.main()
test_sample.py
文件的目录,然后运行以下命令来执行测试:python test_sample.py
或者,如果你使用的是Python 3(推荐),请使用以下命令:
python3 test_sample.py
....
----------------------------------------------------------------------
Ran 4 tests in 0.001s
OK
如果有测试失败,unittest
将显示失败的测试用例及其原因。
你可以根据需要编写更多的测试方法来测试你的代码。只需确保每个测试方法都以test_
开头,并使用unittest.TestCase
提供的断言方法(如assertEqual
、assertTrue
等)来验证你的代码是否按预期工作。