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()