【python】python 测试装饰器模板

待测试代码

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(f"- {response}")

测试用例

test_survey.py

import pytest
from survey import AnonymousSurvey

@pytest.fixture
def language_survey():
	"""⼀个可供所有测试函数使⽤的 AnonymousSurvey 实例"""question = "What language did you first learn to speak?"
	language_survey = AnonymousSurvey(question)
	return language_survey
	
def test_store_single_response(language_survey):
	"""测试单个答案会被妥善地存储"""
	language_survey.store_response('English')
	assert 'English' in language_survey.responses
	
def test_store_three_responses(language_survey):
	"""测试三个答案会被妥善地存储"""
	responses = ['English', 'Spanish', 'Mandarin']
	for response in responses:
		language_survey.store_response(response)
	for response in responses:
		assert response in language_survey.responses

备注：

• 需要导入import pytest
• 使用装饰器@pytest.fixture修饰该函数
• 使用该公共资源的测试函数需要接受一个与装饰器函数同名的形参