example_005_unittest_003.py

"""A simple example to demonstrate Python unit test framework."""

import unittest
from student import Course, Student


class TestStudent(unittest.TestCase):
    """Unit tests for the functions defined in student.py."""

    @classmethod
    def setUpClass(cls) -> None:
        """Performs something once at the beginning of all tests."""
        print("setupClass -- Do something once at the beginning of all tests ")

    @classmethod
    def tearDownClass(cls) -> None:
        """Performs something once at the end of all tests."""
        print("tearDownClass -- Do something once at the end of all tests ")

    def setUp(self):
        """Called before every test case."""
        self.cs313e = Course("CS313E", 4, 3000)
        self.cs329e = Course("CS329E", 4, 2500)

        self.john = Student("John", "Doe", "01/01/1996",
                            [self.cs313e, self.cs329e])
        self.matt = Student("Matt", "Doe", "01/01/1998", [self.cs313e])

        print("Before Every Test Case ")

    def tearDown(self):
        """Called after every test case."""
        # print("For example do something after each test ... ")

    def test_generate_email001(self):
        """Test generating an email with assertEqual."""
        self.assertEqual(self.john.generate_email(
            "utexas.edu"), "John.Doe@utexas.edu")

    def test_generate_email002(self):
        """Test generating an email with assertTrue."""
        self.assertTrue("@" in list(self.john.generate_email("utexas.edu")))

    def test_generate_email003(self):
        """Test generating an email with assertIn."""
        self.assertIn("@", list(self.john.generate_email("utexas.edu")))

    def test_paid_tuition(self):
        """Test generating the paid tuition sum."""
        self.assertEqual(self.john.paid_tuition(), (3000 + 2500))

    def test_cal_discount(self):
        """Tests if the discount calculation is correct with 2 numbers after decimal point."""
        self.assertAlmostEqual(
            self.john.cal_discount(20), (3000 + 2500) * 0.2, 2)


if __name__ == '__main__':
    unittest.main()