python基础===单元测试unittest

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