Python 通过命令行在 unittest.TestCase 中运行单元测试

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)：检查是否抛出指定的异常。