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


class TestTask(unittest.TestCase):

    def test_dataframe_structure(self):
        import user_code
        importlib.reload(user_code)
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.compare_regional_performance()
        output = f.getvalue()
        lines = [l.strip() for l in output.splitlines() if l.strip()]
        all_regions_idx = next((i for i, l in enumerate(lines) if "all regions" in l.lower()), None)
        _dynamic_test(
            self,
            all_regions_idx is not None and len(lines) - all_regions_idx >= 5,
            "DataFrame has correct columns and at least four regions",
            "Expected at least four region rows in the summary table"
        )

    def test_total_sales_calculation(self):
        import user_code
        importlib.reload(user_code)
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.compare_regional_performance()
        output = normalize_text(f.getvalue())
        expected_total = "total sales: 460000"
        _dynamic_test(
            self,
            expected_total in output,
            "Total sales is correctly calculated and displayed",
            f"Expected '{expected_total}' in output, got: '{output}'"
        )

    def test_highest_sales_region(self):
        import user_code
        importlib.reload(user_code)
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.compare_regional_performance()
        output = normalize_text(f.getvalue())
        expected = "highest sales: east (135000)"
        _dynamic_test(
            self,
            expected in output,
            "Highest sales region and value correctly displayed",
            f"Expected '{expected}' in output, got: '{output}'"
        )

    def test_lowest_sales_region(self):
        import user_code
        importlib.reload(user_code)
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.compare_regional_performance()
        output = normalize_text(f.getvalue())
        expected = "lowest sales: south (95000)"
        _dynamic_test(
            self,
            expected in output,
            "Lowest sales region and value correctly displayed",
            f"Expected '{expected}' in output, got: '{output}'"
        )

    def test_all_regions_summary(self):
        import user_code
        importlib.reload(user_code)
        f = io.StringIO()
        with redirect_stdout(f):
            user_code.compare_regional_performance()
        output = normalize_text(f.getvalue())
        for region, sales in [("north", "120000"), ("south", "95000"), ("east", "135000"), ("west", "110000")]:
            _dynamic_test(
                self,
                region in output and sales in output,
                f"Region '{region}' and sales '{sales}' are displayed in summary table",
                f"Expected region '{region}' and sales '{sales}' in output, got: '{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
    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 not assign_nodes:
        return code
    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

Advance your Python automation skills by learning to generate analytical summaries, business reports, and visual dashboards automatically. This course guides you through practical workflows for automating data analysis, reporting, and visualization tasks using Python, pandas, matplotlib, and seaborn.

Learn to automate the generation of analytical summaries, business reports, and visual dashboards using Python. Alternate between concept introduction and hands-on automation tasks for real-world reporting scenarios.

Challenge: Compare Regional Performance

Solution