如何使用Python进行单元测试

开发 后端
在我的日常工作中,我是一名专业程序员。我使用c++、c#和Javascript。我是一个开发团队的一员,他们使用单元测试来验证我们的代码是否按照它应该的方式工作。

 [[390085]]

在本文中,我将通过讨论以下主题来研究如何使用Python创建单元测试。

  • 单元测试基础
  • 可用的Python测试框架
  • 测试设计原则
  • 代码覆盖率

单元测试基础

我使用FizzBuzz编码方式创建了单元测试示例。编码类型是程序员的练习。在这个练习中,程序员试图解决一个特定的问题。但主要目标不是解决问题,而是练习编程。FizzBuz是一个简单的代码类型,非常适合解释和展示Python中的单元测试。

单元测试

单元测试是程序员为测试程序的一小部分而编写的自动化测试。单元测试应该运行得很快。与文件系统、数据库或网络交互的测试不是单元测试。

为了在Python中创建第一个FizzBuzz单元测试,我定义了一个继承自unittest.TestCase的类。这个unittest模块可以在Python的标准安装中获得。

 

  1. import unittest 
  2.  
  3. class FizzBuzzTest(unittest.TestCase): 
  4.  
  5.     def test_one_should_return_one(self): 
  6.         fizzbuzz = FizzBuzz() 
  7.         result = fizzbuzz.filter(1) 
  8.         self.assertEqual('1', result) 
  9.  
  10.     def test_two_should_return_two(self): 
  11.         fizzbuzz = FizzBuzz() 
  12.         result = fizzbuzz.filter(2) 
  13.         self.assertEqual('2', result) 

第一个测试用例验证数字1是否通过了FizzBuzz过滤器,它将返回字符串' 1 '。使用self验证结果。assertEqual方法。方法的第一个参数是预期的结果,第二个参数是实际的结果。

测试用例

我们在测试用例FizzBuzzTest类中调用test_one_should_return_one()方法。测试用例是测试程序特定部分的实际测试代码。

第一个测试用例验证数字1是否通过了FizzBuzz过滤器,它将返回字符串' 1 '。使用self验证结果。assertEqual方法。方法的第一个参数是预期的结果,第二个参数是实际的结果。

如果您查看这两个测试用例,您会看到它们都创建了FizzBuzz类的一个实例。第一个在第6行,另一个在第11行。

我们可以从这两个方法中重构FizzBuzz实例的创建,从而改进代码。

 

  1. import unittest 
  2. class FizzBuzzTest(unittest.TestCase): 
  3.     def setUp(self): 
  4.         self.fizzbuzz = FizzBuzz() 
  5.  
  6.     def tearDown(self): 
  7.         pass 
  8.  
  9.     def test_one_should_return_one(self): 
  10.         result = self.fizzbuzz.filter(1) 
  11.         self.assertEqual('1', result) 
  12.  
  13.     def test_two_should_return_two(self): 
  14.         result = self.fizzbuzz.filter(2) 
  15.         self.assertEqual('2', result) 

我们使用setUp方法创建FizzBuzz类的实例。TestCase基类的设置在每个测试用例之前执行。

另一个方法tearDown是在每个单元测试执行之后调用的。你可以用它来清理或关闭资源。

测试夹具

方法的设置和拆卸是测试夹具的一部分。测试夹具用于配置和构建被测试单元。每个测试用例都可以使用这些通用条件。在本例中,我使用它创建FizzBuzz类的实例。

要运行单元测试,我们需要一个测试运行器。

测试运行器

测试运行程序是执行所有单元测试并报告结果的程序。Python的标准测试运行器可以使用以下命令在终端上运行。

python -m unittest test_fizzbuzz.py

 

 

 

 

测试套件

单元测试词汇表的最后一个术语是测试套件。测试套件是测试用例或测试套件的集合。通常一个测试套件包含应该一起运行的测试用例。

单元测试设计

测试用例应该被很好地设计。考试的名称和结构是最重要的。

测试用例名称

测试的名称非常重要。它就像一个总结考试内容的标题。如果测试失败,你首先看到的就是它。因此,名称应该清楚地表明哪些功能不起作用。

测试用例名称的列表应该读起来像摘要或场景列表。这有助于读者理解被测单元的行为。

构造测试用例方法体

一个设计良好的测试用例由三部分组成。第一部分,安排、设置要测试的对象。第二部分,Act,练习被测单元。最后,第三部分,断言,对应该发生的事情提出主张。

有时,我在单元测试中添加这三个部分作为注释,以使其更清楚。

 

  1. import unittest 
  2.  
  3. class FizzBuzzTest(unittest.TestCase): 
  4.  
  5.     def test_one_should_return_one(self): 
  6.         # Arrange 
  7.         fizzbuzz = FizzBuzz() 
  8.         # Act 
  9.         result = fizzbuzz.filter(1) 
  10.         # Assert 
  11.         self.assertEqual('1', result) 

每个测试用例的单个断言

尽管在一个测试用例中可能有很多断言。我总是尝试使用单个断言。

原因是,当断言失败时,测试用例的执行就会停止。因此,您永远不会知道测试用例中的下一个断言是否成功。

使用pytest进行单元测试

在上一节中,我们使用了unittest模块。Python的默认安装安装这个模块。unittest模块于2001年首次引入。基于Kent Beck和Eric Gamma开发的流行的Java单元测试框架JUnit。

另一个模块pytest是目前最流行的Python单元测试框架。与unittest框架相比,它更具有python风格。您可以将测试用例定义为函数,而不是从基类派生。

因为pytest不在默认的Python安装中,所以我们使用Python的包安装程序PIP来安装它。通过在终端中执行以下命令,可以安装pytest。

pip install pytest

下面我将第一个FizzBuzz测试用例转换为pytest。

 

  1. def test_one_should_return_one(): 
  2.     fizzbuzz = FizzBuzz() 
  3.     result = fizzbuzz.filter(1) 
  4.     assert '1' == result 

有三个不同点。首先,您不需要导入任何模块。其次,您不需要实现一个类并从基类派生。最后,您可以使用标准的Python assert方法来代替自定义的方法。

测试装置

您还记得,单元测试模块使用setUp和tearDown来配置和构建测试中的单元。相反,pytest使用@pytest.fixture属性。在您的测试用例中,您可以使用用该属性装饰的方法的名称作为参数。

pytest框架在运行时将它们连接起来,并将fizzBuzz实例注入测试用例中。

 

  1. @pytest.fixture 
  2. def fizzBuzz(): 
  3.     return FizzBuzz() 
  4.  
  5. def test_one_should_return_one(fizzBuzz): 
  6.     result = fizzBuzz.filter(1) 
  7.     assert result == '1' 
  8.  
  9. def test_two_should_return_two(fizzBuzz): 
  10.     result = fizzBuzz.filter(2) 
  11.     assert result == '2' 

如果您想要模拟单元测试tearDown()方法的行为,可以使用相同的方法来实现。不使用return,而是使用yield关键字。然后,您可以将清理代码放在yield之后。

 

  1. @pytest.fixture 
  2. def fizzBuzz(): 
  3.     yield FizzBuzz() 
  4.     # put your clean up code here 

pytest标记

标记是可以在测试各种函数时使用的属性。例如,如果您将跳过标记添加到您的测试用例中,测试运行器将跳过测试。

 

  1. @pytest.mark.skip(reason="WIP"
  2. def test_three_should_return_fizz(fizzBuzz): 
  3.     result = fizzBuzz.filter(3) 
  4.     assert result == 'Fizz' 

pytest插件生态系统

pytest有很多插件可以添加额外的功能。到我写这篇文章的时候,已经有将近900个插件了。例如,pytest-html和pytest-sugar。

pytest-html

pytest- HTML是pytest的插件,它为测试结果生成HTML报告。当您在构建服务器上运行单元测试时,这非常有用。

pytest-sugar

pytest-sugar改变pytest的默认外观和感觉。它会添加一个进度条,并立即显示失败的测试。

创建代码覆盖率报告

有一些工具可以创建代码覆盖率报告。这个代码覆盖率报告显示了您的单元测试执行了哪些代码。

我使用Coverage和pytest-cov来创建代码覆盖率报告。覆盖率是度量代码覆盖率的通用包。模块pytest-cov是pytest的一个插件,用于连接到Coverage。

都可以使用pip安装。

 

  1. pip install coverage 
  2.  
  3. pip install pytest-cov 

 

在您安装了这两个命令之后,您可以使用这两个命令生成覆盖率报告。在终端或命令中运行它们。

 

  1. coverage run -m pytest 
  2.  
  3. coverage html 

 

第一个生成覆盖率数据。第二个命令将数据转换为HTML报告。Coverage将报告存储在文件系统的htmlcov文件夹中。

如果你在浏览器中打开index.html,它会显示每个文件覆盖率的概览。

 

 

 

 

如果您选择一个文件,它将显示下面的屏幕。覆盖率向源代码添加了一个指示,显示单元测试覆盖了哪一行。

下面我们看到我们的单元测试并没有涵盖第12行和第16行。

 

 

 

 

分支覆盖度量

覆盖率还支持分支覆盖率度量。有了分支覆盖率,如果您的程序中有一行可以跳转到下一行以上,覆盖率跟踪是否访问了这些目的地。

您可以通过执行以下命令来创建带有分支覆盖率的覆盖率报告。

 

  1. pytest——cov-report html:htmlcov——cov-branch——cov=alarm 

我指示pytest生成一个带有分支覆盖的HTML覆盖报告。它应该将结果存储在htmlcov中。而不是为所有文件生成覆盖率报告,我告诉覆盖率只使用alarm.py。

 

 

 

 

责任编辑:华轩 来源: Python学会
相关推荐

2013-06-04 09:49:04

Spring单元测试软件测试

2017-01-14 23:26:17

单元测试JUnit测试

2017-01-16 12:12:29

单元测试JUnit

2017-03-23 16:02:10

Mock技术单元测试

2023-08-02 13:59:00

GoogleTestCTest单元测试

2010-03-04 15:40:14

Python单元测试

2017-12-12 13:17:36

机器学习代码单元测试

2023-12-11 08:25:15

Java框架Android

2023-10-07 08:49:56

测试驱动开发Xunit 框架

2024-10-16 16:09:32

2017-01-14 23:42:49

单元测试框架软件测试

2011-07-27 13:39:28

iOS GH-Unit 单元测试

2024-03-29 08:03:48

单元测试流量

2023-07-26 08:58:45

Golang单元测试

2009-08-19 09:00:48

单元测试框架自动化测试

2012-05-17 09:09:05

Titanium单元测试

2020-09-30 08:08:15

单元测试应用

2011-04-18 13:20:40

单元测试软件测试

2012-11-01 11:32:23

IBMdw

2012-11-01 11:37:05

JavaScript单元测试测试工具
点赞
收藏

51CTO技术栈公众号