11-3

employee.py

class Employee():

    def __init__(self,firstname,lastname,salary):
        self.firstname = firstname
        self.lastname = lastname
        self.salary = salary

    def give_raise(self,increase = 5000):
        self.salary = int(self.salary) + increase

    def display_employee(self):
        fullname = self.firstname + self.lastname
        print(fullname.title() + ':' + str(self.salary))

testemployee.py

import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):

    def setUp(self):
        self.Employee = Employee('Cao','Feng',5000)

    def test_give_default_raise(self):
        self.Employee.give_raise()

    def test_give_custom_raise(self):
        self.Employee.give_raise(10000)

    def test_display_employee(self):
        self.Employee.display_employee()