Python编程：从入门到实践第十一章练习11.2

第十一章练习11.2

#coding:gbk

#11-3雇员
#employee.py
class Employee():
	"""提高工资的雇员信息"""
	
	def __init__(self,firs,last,wage):
		"""定义三项属性"""
		self.first = first
		self.last = last
		self.wage = wage
		
	def give_rasie(self,amount = 5000):
		"""给员工升薪"""
		self.wage += amount


#test_employee.py
import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):
	"""测试Employee能否正常运行"""
	def setUp(self):
		"""创建一个实例"""
		self.john = Employee('john','su',5000)
	
	def test_give_default_raise(self):
		"""测试默认值升薪有效"""
		self.john.give_raise()
		self.assertEqual(self.john.wage,10000)
		
	def test_give_custom_raise(self):
		"""测试非默认值升薪有效"""
		self.john.give_raise(10000)
		self.assertEqual(self.john.wage,15000)
		
unittest.main()