import unittest
from total_quadr import total_quadr
from total_quadr import quadr_total

class ValauesTestCase(unittest.TestCase):
    """Test for total_quadr.py"""
    def test_total_quadr_value(self):
        """The total_quadr function calculates values correct"""
        result = total_quadr(2, 2, 3, 4, 5)
        self.assertEqual(result, 54)

    def test_quadr_total_value(self):
        """The quadr_total function calculates values correct"""
        result = quadr_total(2, 2, 3, 4, 5)
        self.assertEqual(result, 196)


unittest.main()