import unittest
import re
import importlib
from io import StringIO
from contextlib import redirect_stdout

def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)

class TestUserCode(unittest.TestCase):

    def test_multiplication_operation_result_is_correct(self):
        import user_code
        output_capture = StringIO()
        with redirect_stdout(output_capture):
            importlib.reload(user_code)
        output = output_capture.getvalue().strip()
        lines = output.splitlines()

        if len(lines) < 1:
            _dynamic_test(
              self,
              False,
              "",
              "Missing multipication result."
            )

        multiplication_result = lines[0].strip()

        expected_multiplication = str(7.32 * 0.2) 
        condition = multiplication_result == expected_multiplication

        _dynamic_test(
            self,
            condition,
            "Multiplication result is correct.",
            f"Expected multiplication result to be '{expected_multiplication}' but got '{lines[0].strip()}'."
        )

    def test_division_operation_result_is_correct(self):
        import user_code
        output_capture = StringIO()
        with redirect_stdout(output_capture):
            importlib.reload(user_code)
        output = output_capture.getvalue().strip()
        lines = output.splitlines()

        if len(lines) < 2:
            _dynamic_test(
              self,
              False,
              "",
              "Missing division result."
            )

        division_result = lines[1].strip()
        expected_division = str(8.30 / 20)
        condition = division_result == expected_division

        _dynamic_test(
            self,
            condition,
            "Division result is correct.",
            f"Expected division result to be '{expected_division}' but got '{lines[1].strip()}'."
        )

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


Challenge: Multiply & Divide

Solution

Challenge: Multiply & Divide

Solution