class TestWorker(unittest.TestCase):
def setUp(self):
self.worker1 = Employee('Bartosz', 'Chojnacki', 10000)
self.year_sum = [15000, 17000]
def test_give_default_raise(self):
self.worker1.give_raise()
self.assertEqual(self.year_sum[0], self.worker1.year_revenue)
def test_give_custom_raise(self):
self.worker1.give_raise(7000)
self.assertEqual(self.year_sum[1], self.worker1.year_revenue)
class WorkerTestCase(unittest.TestCase):
"""Тесты для Employee."""
def setUp(self):
"""Создание экземпляра Employee и ответов."""
self.worker = Employee('danil', 'lomakin', 120000)
def test_give_default_raise(self):
self.worker.give_raise()
self.assertEqual(self.worker.salary, 125000)
def test_give_custom_raise(self):
self.worker.give_raise(10000)
self.assertEqual(self.worker.salary, 130000)
class TestEmployee(unittest.TestCase):
"""Тесты для класса Employee"""
def setUp(self):
"""Создание работника для использования в тестах"""
self.alex = Employee('alex', 'lutai', 1000)
def test_give_default_raise(self):
"""Тест который проверяет надбавку по умолчанию"""
self.alex.give_raise()
self.assertEqual(self.alex.salary, 6000)
def test_give_custom_raise(self):
"""Тест который проверяет любую надбавку"""
self.alex.give_raise(7000)
self.assertEqual(self.alex.salary, 8000)