Загрузка данных


import unittest


class Employee:
    def __init__(self, name, surname, salary):
        self.name = name
        self.surname = surname
        self.salary = salary

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


class TestEmployee(unittest.TestCase):

    def setUp(self):
        self.emp = Employee('Иван', 'Иванов', 50000)

    def test_give_default_raise(self):
        self.emp.give_raise()
        self.assertEqual(self.emp.salary, 55000)

    def test_give_custom_raise(self):
        self.emp.give_raise(10000)
        self.assertEqual(self.emp.salary, 60000)


unittest.main()