Написать тестовый сценарий используя метод setUp()

class Employee():
    def __init__(self, name, second_name, salary):
        self.name = name 
        self.second_name = second_name
        self.salary = salary
 
    def give_default_raise(self):
        self.salary += 5000
        return self.salary
 
    def give_custom_raise(self):
        request = input("На сколько увеличить зарплату: ")
        self.salary += int(request)
        return self.salary

from unittest import main, TestCase
 
 
class Employee():
    def __init__(self, name, second_name, salary):
        self.name = name
        self.second_name = second_name
        self.salary = salary
 
    def give_raise(self, amount=5000):
        self.salary += amount
 
 
class TestEmployee(TestCase):
    def setUp(self):
        self.employee = Employee('John', 'Smith', 3500)
 
    def test_give_default_raise(self):
        self.employee.give_raise()
        self.assertEqual(8500, self.employee.salary)
 
    def test_give_custom_raise(self):
        self.employee.give_raise(500)
        self.assertEqual(4000, self.employee.salary)
 
 
if __name__ == '__main__':
    main()