
import unittest
import user_code
import ast
import re   
import importlib
import csv
import unittest
import importlib
import io
from contextlib import redirect_stdout

class TestTask(unittest.TestCase):
    def setUp(self):
        self.module_name = "user_code"
        self.test_cases = [
            {"name": "Test Addition", "expected": 5, "actual": 2 + 3},
            {"name": "Test Subtraction", "expected": 3, "actual": 5 - 2},
            {"name": "Test Multiplication", "expected": 9, "actual": 3 * 3},
            {"name": "Test Division", "expected": 2, "actual": 10 // 5},
            {"name": "Test Failure", "expected": 4, "actual": 2 * 2 + 1}
        ]

    def test_each_test_case_pass_fail_recorded(self):
        import user_code
        importlib.reload(user_code)
        # Patch print to capture output
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.run_test_suite(self.test_cases)
        output = f.getvalue()
        # Check that each test case has PASS/FAIL line
        for test in self.test_cases:
            expected = test["expected"]
            actual = test["actual"]
            status = "PASS" if actual == expected else "FAIL"
            line = f"{test['name']}: {status}"
            _dynamic_test(
                self,
                line in output,
                f"Test '{test['name']}' correctly records {status}",
                f"Test '{test['name']}' missing or wrong status line. Output: {output}",
            )

    def test_summary_counts_correct(self):
        import user_code
        importlib.reload(user_code)
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.run_test_suite(self.test_cases)
        output = f.getvalue()
        total = len(self.test_cases)
        passed = sum(1 for t in self.test_cases if t["actual"] == t["expected"])
        failed = total - passed
        _dynamic_test(
            self,
            f"Total: {total}" in output,
            "Summary includes correct total count",
            f"Summary missing or wrong total count. Output: {output}",
        )
        _dynamic_test(
            self,
            f"Passed: {passed}" in output,
            "Summary includes correct passed count",
            f"Summary missing or wrong passed count. Output: {output}",
        )
        _dynamic_test(
            self,
            f"Failed: {failed}" in output,
            "Summary includes correct failed count",
            f"Summary missing or wrong failed count. Output: {output}",
        )

    def test_pass_fail_logic(self):
        import user_code
        importlib.reload(user_code)
        # Test with custom cases
        cases = [
            {"name": "All Pass", "expected": 1, "actual": 1},
            {"name": "All Fail", "expected": 2, "actual": 3},
        ]
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.run_test_suite(cases)
        output = f.getvalue()
        _dynamic_test(
            self,
            "All Pass: PASS" in output and "All Fail: FAIL" in output,
            "Function correctly distinguishes pass/fail for custom cases",
            f"Output missing correct pass/fail for custom cases. Output: {output}",
        )

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)

def normalize_text(text):
    text = text.lower()
    text = re.sub(r"\\s{2,}", " ", text)
    text = re.sub(r"\\s*([,:?])\\s*", r"\\1 ", text)
    return text.strip()

def change_var(code: str, var_name: str, value: str) -> str:
    tree = ast.parse(code)
    lines = code.splitlines()
    changed = False
    # Collect all assignment nodes to modify
    assign_nodes = [
        (i, node)
        for i, node in enumerate(tree.body)
        if isinstance(node, ast.Assign)
        and any(isinstance(target, ast.Name) and target.id == var_name for target in node.targets)
    ]

    # If nothing to change, return unmodified code
    if not assign_nodes:
        return code

    # Perform replacements for all matching assignments (from last to first to not break line offsets)
    for i, node in reversed(assign_nodes):
        start_line = node.lineno - 1
        line = lines[start_line]
        indent = ' ' * (len(line) - len(line.lstrip()))
        lines[start_line] = f"{indent}{var_name} = {value}"
        next_line = len(lines)
        for next_node in tree.body[i+1:]:
            if hasattr(next_node, 'lineno'):
                next_line = next_node.lineno - 1
                break
        if next_line > start_line + 1:
            lines[start_line+1:next_line] = []
        changed = True

    return '\\n'.join(lines) if changed else code

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


test_main.py

Master Python as a powerful tool for automation engineering. This course guides you through practical and advanced Python techniques to automate repetitive tasks, analyze data, and interact with systems, all tailored for automation engineers. Each section blends engaging theory with hands-on, real-world challenges.

Learn how Python can automate repetitive and time-consuming tasks commonly faced by automation engineers, from log parsing to report generation.

Dive into data manipulation and analysis using Python and pandas, tailored for automation engineers handling structured data.

Master Python techniques for interacting with systems, automating tests, and analyzing results—core skills for automation engineers.

Challenge: Automate Test Suite

Solution

Challenge: Automate Test Suite

Solution