【Python编程入门】Lecture 11：测试代码

最新推荐文章于 2024-07-19 16:36:18 发布

Jasssssen

最新推荐文章于 2024-07-19 16:36:18 发布

阅读量167

点赞数

分类专栏： Python入门文章标签： python 单元测试

本文链接：https://blog.csdn.net/weixin_44599157/article/details/125544064

版权

Python入门专栏收录该内容

11 篇文章 3 订阅

订阅专栏

11 测试代码

11.1 Python测试框架简介
11.2 举例

11.1 Python测试框架简介

编写函数和类时，还可以为其编写测试，以此确保代码面对各种输入都能按要求工作。

11.1.1 单元测试

单元测试是指对软件中的最小可测试单元进行检查和验证。对于单元测试中单元的含义，一般来说要根据实际情况去判定其具体含义，如C语言中单元指一个函数，Java里单元指一个类等等。
测试用例是一组单元测试，它们用以核实函数在各种情形下的行为的都符合要求。
全覆盖测试是一整套单元测试，包含了各种可能的函数使用方式。

11.1.2 unittest模块

unittest模块是Python标准库中提供的单元测试框架，它设计的灵感来源于Junit，具有和Junit类似的结构。unittest包含以下4个主要成员：

1、TestCase
是测试用例类的父类，通过对其继承，使用户自定义的子类具备执行测试的能力。

2、TestSuite
3、TestLoader
4、TestRunner 和 TextTestRunner

自行参考：https://blog.csdn.net/weixin_45594025/article/details/120362401

11.1.3 断言方法

unittest.TestCase类具有的断言方法是非常有用的，断言方法可以用于核实得到的结果是否与期望的结果一致。如果一致，则程序行为符合预期，如果不一致，Python将引发异常。常用的断言方法如下：

方法	用途
assertEqual(a,b)	核实a==b
assertNotEqual(a,b)	核实a!=b
asssertTrue(a)	核实a为True
assertFalse(a)	核实a为False
assertIn(item,list)	核实item在list中
asserNotIn(item,list)	核实item不在list中

11.1.4 运行测试

unittest.main()是用于启动测试的函数，可以调用该方法来开始测试，执行 unittest.main()会运行以test开头的方法。需要注意的是，很多测试框架都会先导入测试文件再运行，导入时解释器会在导入的同时执行它。需要运用if代码块检查特殊变量__name__，如果这个文件作为主程序执行时，该变量为"__main__"，此时才会执行测试。以下举一个测试函数的例子：

import unittest

def get_formatted_name(first_name, last_name): #需要测试的函数
    """返回整洁的姓名"""
    full_name = f"{first_name.title()} {last_name.title()}"
    return full_name

class NameTestCase(unittest.TestCase):   #该类必须继承unittest.TestCase类
    """测试函数"""
    def test_first_last_name(self): #该方法必须以test_开头
        """是否能够正确地处理只有名和姓的姓名"""
        full_name = get_formatted_name('kevin', 'durant')
        self.assertEqual(full_name, 'Kevin Durant')

if __name__ == '__main__':
    unittest.main()

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

11.2 举例

11.2.1 测试函数未通过的情形

上一个例子是通过测试的情况，如果需要测试的函数不能通过测试，会有如下反馈：

F
======================================================================
FAIL: test_first_last_name (__main__.NameTestCase)
是否能够正确地处理只有名和姓的姓名
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:/……/ch11_1.py", line 13, in test_first_last_name
    self.assertEqual(full_name, 'Kevin Durant')
AssertionError: 'KevinDurant' != 'Kevin Durant'
- KevinDurant
+ Kevin Durant
?      +


----------------------------------------------------------------------
Ran 1 test in 0.000s

FAILED (failures=1)

注意，运行测试用例时，每完成一个单元测试，Python都打印一个字符：测试通过时打印一个句点（.），测试引发错误时打印一个E，测试断言失败是打印一个F，通过观察这些结果可以得知有多少用例通过测试。

11.2.2 添加多个测试用例

如果要测试是否能够处理含有中间名的姓名，可以再编写一个test_开头的方法。同时我修改了被测试函数，使得它能够处理含有中间名的姓名。

import unittest

def get_formatted_name(first_name, last_name, middle_name=''):
    """返回整洁的姓名"""
    if middle_name :
        full_name = f"{first_name.title()} {middle_name.title()} {last_name.title()}"
    else:
        full_name = f"{first_name.title()} {last_name.title()}"
    return full_name

class NameTestCase(unittest.TestCase):   #该类必须继承unittest.TestCase类
    """测试函数"""
    
    def test_first_last_name(self): #该方法必须以test_开头
        """是否能够正确地处理只有名和姓的姓名"""
        full_name = get_formatted_name('kevin', 'durant')
        self.assertEqual(full_name, 'Kevin Durant')
    def test_first_last_middle_name(self):
        full_name = get_formatted_name('wolfgang','mozart','amadeus')
        self.assertEqual(full_name, 'Wolfgang Amadeus Mozart')

if __name__ == '__main__':
    unittest.main()  # 让Python运行这个文件中的测试

..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

11.2.3 测试类的例子

编写一个匿名调查问卷类，包含一个构造方法和三个普通方法，然后运用一下这个类。

import unittest

class AnonymousSurvey: #需要被测试的类
    '''收集匿名调查问卷的答案'''
    
    def __init__(self, question):
        '''存储一个问题，并为存储答案做准备'''
        self.question = question
        self.responses = []
        
    def show_question(self):
        '''显示调查问卷'''
        print(self.question)
    
    def store_response(self, new_response):
        '''存储单份调查答卷'''
        self.responses.append(new_response)
    
    def show_results(self):
        '''显示收集到的所有答卷'''
        print('Survey results:')
        for response in self.responses:
            print(f"-{response}")

#定义一个问题，创建一个调查
question = "What's your favorite color?"
my_survey = AnonymousSurvey(question)

#显示问题并存储答案
my_survey.show_question()
print("Enter 'q' at any time to quit.\n")
while True:
    response = input('Color:')
    if response  == 'q':
        break
    my_survey.store_response(response)

#显示调查结果
print("\nThank you to everyone who particiated in the survey!")
my_survey.show_results()

What's your favorite color?
Enter 'q' at any time to quit.


Color:Blue

Color:q

Thank you to everyone who particiated in the survey!
Survey results:
-Blue

编写测试代码，首先测试一个答案是否会被妥善存储。

import unittest

class AnonymousSurvey: #需要被测试的类
	--snip--

class TestAnonymousSurvey(unittest.TestCase):
    '''针对匿名调查类的测试'''
    def test_store_single_response(self):
        '''测试单个答案会被妥善储存'''
        question = "What's your favorite color?"
        my_survey = AnonymousSurvey(question)
        my_survey.store_response("Blue")
        self.assertIn("Blue", my_survey.responses)

if __name__ == "__main__":
    unittest.main()

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

再编写一个测试两个答案是否能够被妥善储存的测试。

import unittest

class AnonymousSurvey: #需要被测试的类
	--snip--

class TestAnonymousSurvey(unittest.TestCase):
    '''针对匿名调查类的测试'''
    def test_store_single_response(self):
        '''测试单个答案会被妥善储存'''
        question = "What's your favorite color?"
        my_survey = AnonymousSurvey(question)
        my_survey.store_response("Blue")
        self.assertIn("Blue", my_survey.responses)
    
    def test_store_two_responses(self):
    	'''测试两个答案会被妥善存储'''
        question = "What's your favorite color?"
        my_survey = AnonymousSurvey(question)
        my_survey.store_response("Blue")
        my_survey.store_response("Red")
        self.assertIn("Blue", my_survey.responses)
        self.assertIn("Red", my_survey.responses)

if __name__ == "__main__":
    unittest.main()

..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

11.2.4 结合使用setUp()方法

在前面的测试中，每个测试中都创建了一个匿名调查类的实例，并且在每个测试中都创建了答案。TestCase类中的setUp()方法可以让我们只创建一次实例和答案就在后面的每个测试中使用。setUp()方法中需要做两件事：第一是创建一个调查对象，第二是创建一个答案列表。这两样东西存储在属性中，所以存储这两样东西的变量包含前缀self，并且可以在任何地方使用。

import unittest

class AnonymousSurvey: #需要被测试的类
	--snip--

class TestAnonymousSurvey(unittest.TestCase):
    '''针对匿名调查类的测试'''
    
    def setUp(self):
        '''创建一个调查对象和一组答案供后面的测试方法使用'''
        question = "What's your favorite color?"
        self.my_survey = AnonymousSurvey(question)
        self.responses = ["Blue", "Red", "Green"]
        
    def test_store_single_response(self):
        '''测试单个答案会被妥善储存'''
        self.my_survey.store_response(self.responses[0])
        self.assertIn(self.responses[0], my_survey.responses)
    
    def test_store_two_responses(self):
        '''测试两个答案会被妥善存储'''
        for response in self.responses[0:1]: #此处直接用for循环，上个程序也可以改成for循环
            self.my_survey.store_response(response)
        for response in self.responses[0:1]: 
            self.assertIn(response, my_survey.responses)

if __name__ == "__main__":
    unittest.main()