Python编程 从入门到实践 练习11-3

11-3 雇员

employee.py

# 雇员

# 编写一个名为Employee的类，其方法__init__()接受名、姓和年薪，并将他们都存储到属性中。
class Employee():
	def __init__(self, first_name, last_name, annul_salary):
		self.first_name = first_name
		self.last_name = last_name
		self.annul_salary = annul_salary
	
	# 编写一个名为give_raise()的方法，它默认将年薪增加5000美元，但也能够接受其他的年薪
	# 增加量
	def give_raise(self, add_annul_salary = 5000):
		self.annul_salary = self.annul_salary + add_annul_salary

test_employee.py

import unittest
from employee import Employee

# 为Employee编写一个测试用例，其中包含两个测试方法：test_give_default_raise()和
# test_give_custom_raise().
class TestEmployee(unittest.TestCase):
	# 使用方法setUp()，以免在每个测试方法中都创建新的雇员实例
	def setUp(self):
		first_name = 'albert'
		last_name = 'einstein'
		annul_salary = 10000
		self.my_employee = Employee(first_name, last_name, annul_salary)
	
	def test_give_default_raise(self):
		self.my_employee.give_raise()
		self.assertEqual(self.my_employee.annul_salary, 15000)
	
	def test_give_custom_raise(self):
		self.my_employee.give_raise(20000)
		self.assertEqual(self.my_employee.annul_salary, 30000)

unittest.main()

运行这个测试用例，确认两个测试都通过了。