python入门代码-Python入门学习系列——Python代码测试

Python代码测试

对于编写的代码，可以使用unittest模块中的相关方法进行测试。

测试函数

首先定义一个简单的函数，该函数用来合并名称并返回。

name_function.py：

def get_fromatted_name(first,last):

"""组合姓名并返回"""

full_name=first+' '+last

return full_name.title()

接着编写测试代码，需要先导入unittest模块。然后创建一个测试类，用于包含对上述函数的单元测试，这个类必须继承unittest.TestCase类。

test_name_function.py：

#导入模块和要测试的函数

import unittest

from name_function import get_fromatted_name

#编写测试类，必须继承自unittest.TestCase

class NamesTestCase(unittest.TestCase):

"""测试name_function.py"""

#测试方法

def test_first_last_name(self):

"""能够正确的处理Python Java这样的名字吗？"""

formatted_name=get_fromatted_name('python','java')

#调用断言方法用来核实得到的结果是否与期望值结果一致

self.assertEqual(formatted_name,'Python Java')

#运行测试

unittest.main()

上述中最重要的是调用TestCase类的assertEqual()断言方法，用来核实代码执行结果和预期结果是否一致。

注意：当编写测试类时，除了需要继承unittest.TestCase外，测试类中的测试方法必须以test_开头，这样在调用unittest.main()方法时，这些以test_开头的方法才会自动运行。

unittest.TestCase类的断言方法

断言方法检查你认为应该满足的条件是否确实满足。如果该条件确实满足，你对程序行为的假设就得到了确认，你就可以确信其中没有错误。如果你认为应该满足的条件实际上并不满足，Python将引发异常。

方法

用途

assertEqual(a,b)

核实a==b

assertNotEqual(a,b)

核实a!=b

assertTrue(x)

核实x为True

assertFalse(x)

核实x为False

assertIn(item,list)

核实item在list中

assertNotIn(item,list)

核实item不在list中

测试类

首先定义一个简单的类，接着对这个类编写测试代码。

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("答案：")

for response in self.responses:

print('-'+response)

测试类和测试函数类似，核心都是应用断言方法来实现的。

import unittest

from survey import AnonymousSurvey

class TestAnonyousSurvey(unittest.TestCase):

"""对类AnonymousSurvey进行测试"""

def setUp(self):

question="你对什么语言感兴趣？"

self.my_survey=AnonymousSurvey(question)

#设定程序回答的答案

self.responses=["java","python","c#"]

#方法以test_开头

def test_store_three_response(self):

for response in self.responses:

self.my_survey.store_response(response)

for response in self.responses:

self.assertIn(response,self.my_survey.responses)

unittest.main()

setUp()方法

unittest.TestCase类包含方法setUp()，让我们只需创建这些对象一次，就可以在每个测试方法中使用它们。

如果你在自己的测试类中重写了setUp()方法，python将先运行它，再运行各个以test_开头的方法。这样，在你编写的每个测试方法中都可使用在方法setUp()中创建的对象了。

import unittest

from survey import AnonymousSurvey

class TestAnonyousSurvey(unittest.TestCase):

"""对类AnonymousSurvey进行测试"""

def setUp(self):

question="你对什么语言感兴趣？"

self.my_survey=AnonymousSurvey(question)

#设定程序回答的答案

self.responses=["java","python","c#"]

#方法以test_开头

def test_store_three_response(self):

for response in self.responses:

self.my_survey.store_response(response)

for response in self.responses:

self.assertIn(response,self.my_survey.responses)

unittest.main()

总结

python代码单元测试，需要使用unittest模块，尤其是其中的TestCase类及其断言方法。

在编写测试代码时，需要自定义一个测试类，并且继承自unittest.TestCase。

所有编写的测试方法都必须以test_开头，这样在调用unittest.main()方法后，才能自动执行。

如果需要在不同的测试方法中使用同一个对象，可以考虑在类中重写setUp()。

断言方法用于验证程序运行的值和期望值（一般是写死的值）是否一致，从而反馈测试的结果。

参考资源

《Python编程：从入门到实践》

本文后续会随着知识的积累不断补充和更新，内容如有错误，欢迎指正。

最后一次更新时间：2018-08-23