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()