在Ubuntu下进行Python单元测试,你可以使用Python的内置模块unittest或者第三方库pytest
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
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标记,以及详细的错误信息。