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()
运行这个测试用例,确认两个测试都通过了。