python编程从入门到实践11单元测试

单元测试用于测试某个函数是否符合要求，要测试函数需要导入模块unittest和要测试的函数，在创建一个继承unittest.TestCase类。
比如要测试下面get_formatted函数，来检测这个函数能否正确的工作：

def get_formatted_name(first,last):
	full_name=first+" "+last
	return full_name.title()

然后编写一个测试用例：

import unitest
from name_function import get_formatted_name

class NamesTestCase(unittest.TestCase):
	def test_first_last_name(self):
		formatted_name=get_formatted_name('janis','joplin')
		self.assertEqual(formatted_name,'Janis Jopin')
unittest.main()

运行上面的测试程序，如果没有错误信息，则说明测试通过，如果用提示“E“这个信息，则说明有错误，测试没有通过。

如果测试没有通过，则需要检查被测试的函数有没有问题。

我们更改一下上面被测试的函数，改成下面的样子：

def get_formatted_name(first,middle,last):
	full_name=first+" " + middle +" " + last
	return full_name.title()

然后再运行测试函数，则发现提示”E“错误，因为这个被测试函数增了中间的名字，所以导致测试不通过。
中间的middle是可选参数，所以挪到后面，

def get_formatted_name(first,last,middle=''):
	if middle:
		full_name=first+''+middle+‘’+last
	else:
		full_name=first+''+last
	return full_name.title()

添加新的测试

import unitest
from name_function import get_formatted_name

class NamesTestCase(unittest.TestCase):
	def test_first_last_name(self):
		formatted_name=get_formatted_name('janis','joplin')
		self.assertEqual(formatted_name,'Janis Jopin')
	def test_first_last_middle_name(self):
		formatted_name=get_formatted_name('wolfgang','mozart','amadeus')
		self.assertEqual(formatted_name,"Wolfgang Amadeus Mozart")
unittest.main()

我们运行测试函数，没有返回错误，测试通过。

..
Ran 2tests in 0.000s
OK

刚我们了解了函数的测试，接下来了解类的测试。
首先编写一个要测试的类，其文件命名为survey.py.

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('-'+response)

测试AnonymousSurvey类，用assertIn（）来测试：

import unittest
from survey import AnonymousSurvey

class TestAnonmyousSurvey(unittest.TestCase):
	def test_store_single_response(self):
		question="what language did you fiset learn to speak?"
		my_survey=AnonymousSurvey(question)
		my_survey.store_response('English')
		self.assertIn('English',my_survey.responses)
unitest.main()

运行上面的测试后如果返回下面界面说明测试通过：

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

上面测试了一个答案，如果测试三个答案的话就需要把测试函数更改一下：

import unittest
from survey import AnonymousSurvey

class TestAnonmyousSurvey(unittest.TestCase):
	def test_store_single_response(self):
		question="what language did you fiset learn to speak?"
		my_survey=AnonymousSurvey(question)
		my_survey.store_response('English')
		self.assertIn('English',my_survey.responses)
	def test_store_three_responses(self):
		question = "what language did you fiset learn to speak?"
		my_survey=AnonymousSurvey(question)
		responses=['english','spanish','mandarin']
		for response in responses:
			my_survey.store_reponse(response)
		for response in responses:
			self.assertIn(response,my_survey.response)
unitest.main()

python中的setUp()方法可以创建一组对象和答案，为测试提供便利。