文章目录
- unittest 模块简介
- 编写单元测试
- 在命令行中运行所有测试
- 在命令行中运行单个测试
- 使用装饰器跳过测试
- 总结
- 常用断言方法
unittest 模块简介
unittest是Python标准库中的一个模块，用于编写和运行单元测试。它提供了一个单元测试框架，使得编写测试用例、运行测试、进行断言等操作变得非常简单。通过使用unittest，我们能够更容易地验证我们编写的代码是否按照预期工作。
编写单元测试
首先，让我们编写一个简单的测试用例作为示例。我们创建一个名为test_calculator.py的文件，并编写以下代码：
import unittestclass CalculatorTestCase(unittest.TestCase):def test_addition(self):result = 2 + 2self.assertEqual(result, 4)def test_subtraction(self):result = 5 - 3self.assertEqual(result, 2)
上述代码是一个简单的计算器测试用例。我们定义了两个测试方法test_addition和test_subtraction，分别测试了加法和减法操作。
在命令行中运行所有测试
我们可以使用python -m unittest命令来运行所有测试用例。在命令行中进入test_calculator.py所在的目录，并执行以下命令：
python -m unittest test_calculator.py
执行上述命令后，unittest会自动发现并运行CalculatorTestCase中的所有测试方法。在命令行中，我们可以看到测试的结果，包括每个测试方法是否通过、失败或错误等。这是一个非常简单且方便的方法来运行所有测试用例。
在命令行中运行单个测试
有时候，我们只想运行单个测试方法而不是全部测试。unittest提供了一个选项来指定要运行的测试方法。我们可以使用-k选项后跟测试方法的名称来运行单个测试。
下面是运行单个测试的示例命令：
python -m unittest -k test_addition test_calculator.py
执行上述命令后，unittest只会运行名为test_addition的测试方法。这在调试特定问题或运行仅关注的测试时非常有用。
使用装饰器跳过测试
有时候，我们希望跳过某些测试。unittest提供了一个装饰器@unittest.skip("reason")来指定跳过测试的原因。我们可以将它应用到测试方法上，来跳过特定的测试。
下面是一个跳过测试的示例代码：
import unittestclass CalculatorTestCase(unittest.TestCase):@unittest.skip("Skipped test")def test_addition(self):result = 2 + 2self.assertEqual(result, 4)def test_subtraction(self):result = 5 - 3self.assertEqual(result, 2)
在上述代码中，我们使用了@unittest.skip装饰器来跳过test_addition方法。
总结
在本文中，我们介绍了如何通过命令行在 unittest.TestCase 中运行单个测试。我们首先了解了unittest模块的基本知识，并且编写了一个简单的测试用例。然后，我们学习了如何在命令行中运行所有测试和单个测试。最后，我们还了解了如何使用装饰器跳过测试。
unittest模块为我们提供了一个强大且灵活的测试框架，使得编写、运行和组织测试变得更加简单。通过使用unittest，我们可以更加自信地验证我们编写的代码是否按照预期工作。希望本文能对你理解和使用unittest模块有所帮助。
常用断言方法
assertEqual(a, b)：检查a和b是否相等。assertNotEqual(a, b)：检查a和b是否不相等。assertTrue(x)：检查x是否为True。assertFalse(x)：检查x是否为False。assertIsNone(x)：检查x是否为None。assertIsNotNone(x)：检查x是否不为None。assertRaises(exception)：检查是否抛出指定的异常。