''' 编写一个名为Employee 的类,其方法__init__()接受名、姓和年薪,并 将它们都存储在属性中。编写一个名为give_raise()的方法,它默认将年薪增加5000 美元,但也能够接受其他的年薪增加量。 为Employee 编写一个测试用例,其中包含两个测试方法:test_give_default_ raise()和test_give_custom_raise()。使用方法setUp(),以免在每个测试方法中都创 建新的雇员实例。运行这个测试用例,确认两个测试都通过了。 ''' class Employee(): def __init__(self, firstname , lastname, money): self.firstname = firstname self.lastname = lastname self.money = money def give_reise(self,raise=5000): self.money += int(raise) return self.money
import unittest from Employee import Employee class EmployeeTestCase(unittest.TestCase): def setUp(self): ''' 创建新的员工 ''' self.new_a = Employee("li", "ming", 700) self.new_b = Employee("xiao", "hong", 2000) self.new_c = Employee("WWW", "COM", 5000) def test_give_default_raise(self): self.new_a.give_reise() x = self.new_a.money print(x) self.assertEqual(x,5700) def test_give_custom_reise(self): self.new_b.give_reise(1000) self.new_c.give_reise(8000) self.assertEqual(self.new_b.money, 3000) self.assertEqual(self.new_c.money, 13000) unittest.main()