employee_info.py

class Employee:

    """雇员信息"""

    def __init__(self, first, last, year_salary):

        """属性"""

        self.first_name = first

        self.last_name = last

        self.year_salary = year_salary

    def give_raise(self, add_year_salary=5000):

        """年薪增加量"""

        self.year_salary = self.year_salary + add_year_salary

        return self.year_salary

test_employee_info.py

import unittest

from employee_info import Employee

class TestEmployee(unittest.TestCase):

    """针对Employee类的测试"""

    def setUp(self):

        """

        创建雇员实例供测试方法使用

        """

        self.employee = Employee('bob', 'li', 0)

    def test_give_default_raise(self):

        """测试默认年薪加量"""

        self.default_raise = self.employee.give_raise()

        self.assertEqual(5000, self.default_raise)

    def test_give_custom_raise(self):

        """测试其他的年薪增加量"""

        self.add_raise = 8000

        self.custom_raise = self.employee.give_raise(self.add_raise)

        self.assertEqual(self.add_raise, self.custom_raise)

if __name__ == "__main__":

    unittest.main()