'''

编写一个名为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()