个人答案,仅供参考
EmployeeSurver.py
class Employee():
def __init__(self,firstname,lastname,salary):
"""默认方法"""
self.firstname = firstname
self.lastname = lastname
self.salary = salary
def give_raise(self,up=''):
"""增加年薪,默认增加$5000"""
if up:
self.salary += up
else:
self.salary += 5000
return self.salary
test_employee.py
import unittest
from EmployeeSurver import Employee
class TestEmployee(unittest.TestCase):
"""针对Employee的测试"""
def setUp(self):
"""创建测试对象"""
self.employeeTest = Employee('Harper','Zeng',10000)
def test_give_default(self):
self.employeeTest.give_raise()
self.assertEqual(self.employeeTest.salary,15000)
def test_give_custom_raise(self):
self.employeeTest.give_raise(10000)
self.assertEqual(self.employeeTest.salary,20000)
unittest.main()
运行效果: