ubuntu

如何在Ubuntu下进行Python单元测试

小樊
38
2025-10-18 14:22:27
栏目: 编程语言

在Ubuntu下进行Python单元测试,你可以使用Python的内置模块unittest或者第三方库pytest

  1. 使用Python内置的unittest模块:

首先,创建一个Python文件,例如test_my_module.py,然后在该文件中编写测试用例。这里是一个简单的例子:

import unittest
from my_module import add

class TestMyModule(unittest.TestCase):

    def test_add(self):
        self.assertEqual(add(1, 2), 3)
        self.assertEqual(add(-1, 1), 0)
        self.assertEqual(add(0, 0), 0)

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

在这个例子中,我们导入了unittest模块和我们要测试的my_module模块。然后,我们创建了一个继承自unittest.TestCase的测试类TestMyModule,并定义了一个测试方法test_add。在这个方法中,我们使用assertEqual方法来检查add函数的输出是否符合预期。

最后,我们在文件的末尾添加了if __name__ == '__main__': unittest.main(),这样就可以直接运行这个测试文件了。

在终端中,使用以下命令运行测试:

python3 test_my_module.py
  1. 使用第三方库pytest

首先,你需要安装pytest。在终端中运行以下命令:

pip3 install pytest

然后,创建一个Python文件,例如test_my_module.py,并编写测试用例。这里是一个简单的例子:

from my_module import add

def test_add():
    assert add(1, 2) == 3
    assert add(-1, 1) == 0
    assert add(0, 0) == 0

在这个例子中,我们导入了my_module模块,并定义了一个测试函数test_add。在这个函数中,我们使用assert语句来检查add函数的输出是否符合预期。

最后,在终端中,使用以下命令运行测试:

pytest test_my_module.py

pytest会自动发现并运行所有以test_开头的函数。如果所有测试都通过,你将看到一个绿色的PASSED标记。如果有测试失败,你将看到一个红色的FAILED标记,以及详细的错误信息。

0
看了该问题的人还看了