在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
模块。然后,我们创建了一个TestMyModule
类,该类继承自unittest.TestCase
。在这个类中,我们定义了一个名为test_add
的测试方法,该方法使用assertEqual
来检查add
函数的输出是否符合预期。
最后,我们在文件的末尾添加了if __name__ == '__main__': unittest.main()
,这样我们就可以直接运行这个文件来执行测试。
在终端中,导航到包含test_my_module.py
的目录,然后运行以下命令:
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
函数的输出是否符合预期。
最后,在终端中,导航到包含test_my_module.py
的目录,然后运行以下命令:
pytest test_my_module.py
这将运行test_my_module.py
文件中的所有测试函数,并显示测试结果。