【py code everyday】2023-02-16

第11章
针对上一个内容，核实当提供3个答案时，也能妥善处理
修改下述代码

import unittest
from test import AnonymousSurvey

class TestAnonymousSurvey(unittest.TestCase):
	#针对AnonymousSurvey类的测试
	def test_store_single_response(self):
		#测试单个答案被妥善地存储
		question = "What language did you first learn to speak?"
		my_survey = AnonymousSurvey(question)
		my_survey.store_response('English')
		self.assertIn('English',my_survey.responses)
if __name__ == '__main__':
	unittest.main()

import unittest
from test import AnonymousSurvey

class TestAnonymousSurveySurvey(unittest.TestCase):
	def test_store_single_response(self):
		question = "What language did you first 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 first learn to speak?"
		my_survey = AnonymousSurvey(question)
		responses= ['English', 'Chinese', 'Korean']
		for response in responses:
			my_survey.store_response(response)

		for response in responses:
			self.assertIn(response, my_survey.responses)
if __name__ == '__main__':
	unittest.main()

responses 用列表来存储3个答案，再对每个答案都调用my_survey.store_response（）
存储后，再使用一个循环来确认每个答案都包含在my_survey.responses

..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK
[Finished in 109ms]

上述做法有些重复的地方
下面使用unittest的另一项功能来提高效率
使用setUp（），这样在TestCase类中，Python将先运行它，再运行各个以test_开头的方法
也就是说，下面将使用setUp（）来创建一个调查对象和一组答案，供2个实例使用

import unittest
from test import AnonymousSurvey

class TestAnonymousSurveySurvey(unittest.TestCase):

	def setUp(self):
		#创建一个调查对象和一组答案，供使用的测试方法使用
		question = "What language did you first learn to speak?"
		self.my_survey = AnonymousSurvey(question)
		self.responses = ['English','Chinese','Korean']

	def test_store_single_response(self):
		#测试单个答案被存储
		self.my_survey.store_response(self.responses[0])
		self.assertIn(self.responses[0], self.my_survey.responses)

	def test_store_three_responses(self):
		#测试多个答案被存储
		for response in self.responses:
			self.my_survey.store_response(response)
		for response in self.responses:
			self.assertIn(response,self.my_survey.responses)
if __name__ == '__main__':
	unittest.main()

方法 setUp()做了2件事，创建一个调查对象，和 创建一个答案列表。
存储这两样东西的变量名包含前缀self，即存储在属性中，因此这个类可以在任何地方使用，这样测试方法更简单，不需要再单独创建调查对象和答案了
test_store_single_response() 核实self.responses 中的第一个答案self.responses[0] 被妥善地存储，而方法
test_store_three_response() 核实self.responses 中的 全部三个答案 都被妥善地存储

编写一个名为 Employee 的类，其方法 init() 接受名、姓和年薪，并将它们都存
储在属性中。编写一个名为 give_raise() 的方法，它默认将年薪增加 5000 美元，但也能
够接受其他的年薪增加量。
为 Employee 编写一个测试用例，其中包含两个测试方法：test_give_default_raise()
和 test_give_custom_raise() 。使用方法 setUp() ，以免在每个测试方法中都创建新的雇
员实例。运行这个测试用例，确认两个测试都通过了。

class Employee:
    def __init__(self,first,last,payment):
        self.first = first
        self.last = last
        self.payment = payment

    def give_raise(self,amount=5000):
        self.payment += amount

import unittest
from homework import Employee

class TestEmployee(unittest.TestCase):
	def setUp(self):
		self.eric = Employee('eric','maty',2000)

	def test_give_default_raise(self):
		self.eric.give_raise()
		self.assertEqual(self.eric.payment,7000)
	def test_give_custom_raise(self):
		self.eric.give_raise(10000)
		self.assertEqual(self.eric.payment, 12000)


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

在这里使用assertEqual方法，self.erci相当于一个实参