
import unittest
import user_code
import ast
import re   
import unittest
import importlib
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import sys
import types

class TestTask(unittest.TestCase):
    def setUp(self):
        # Patch plt.show to prevent actual plot display
        self._original_show = plt.show
        plt.show = lambda: None
        import user_code
        importlib.reload(user_code)

    def tearDown(self):
        plt.show = self._original_show
        plt.clf()

    def test_bar_chart_seven_bars(self):
        import user_code
        plt.clf()
        expenses = [10, 20, 30, 40, 50, 60, 70]
        user_code.plot_weekly_spending(expenses)
        bars = [child for child in plt.gca().get_children() if isinstance(child, matplotlib.patches.Rectangle)]
        # The first bar is the background, so skip it
        bar_count = len([b for b in bars if b.get_x() != 0 or b.get_y() != 0 or b.get_width() != 1 or b.get_height() != 1])
        _dynamic_test(
            self,
            bar_count == 7,
            "Bar chart has 7 bars for 7 days of the week",
            f"Expected 7 bars, found {bar_count} bars."
        )

    def test_x_axis_label(self):
        import user_code
        plt.clf()
        user_code.plot_weekly_spending([1,2,3,4,5,6,7])
        xlabel = plt.gca().get_xlabel()
        _dynamic_test(
            self,
            xlabel == "Day of the Week",
            "X-axis label is correctly set to 'Day of the Week'",
            f"Expected x-axis label 'Day of the Week', got '{xlabel}'"
        )

    def test_y_axis_label(self):
        import user_code
        plt.clf()
        user_code.plot_weekly_spending([1,2,3,4,5,6,7])
        ylabel = plt.gca().get_ylabel()
        _dynamic_test(
            self,
            ylabel == "Amount Spent ($)",
            "Y-axis label is correctly set to 'Amount Spent ($)'",
            f"Expected y-axis label 'Amount Spent ($)', got '{ylabel}'"
        )

    def test_chart_title(self):
        import user_code
        plt.clf()
        user_code.plot_weekly_spending([1,2,3,4,5,6,7])
        title = plt.gca().get_title()
        _dynamic_test(
            self,
            title == "Weekly Spending Overview",
            "Chart title is correctly set to 'Weekly Spending Overview'",
            f"Expected chart title 'Weekly Spending Overview', got '{title}'"
        )

    def test_expenses_order(self):
        import user_code
        plt.clf()
        test_expenses = [11, 22, 33, 44, 55, 66, 77]
        user_code.plot_weekly_spending(test_expenses)
        bars = [child for child in plt.gca().get_children() if isinstance(child, matplotlib.patches.Rectangle)]
        # Filter out the background rectangle (patch)
        filtered_bars = [b for b in bars if b.get_x() != 0 or b.get_y() != 0 or b.get_width() != 1 or b.get_height() != 1]
        heights = [b.get_height() for b in filtered_bars]
        _dynamic_test(
            self,
            heights == test_expenses,
            "Bar heights match the expenses in the order given",
            f"Expected bar heights {test_expenses}, got {heights}"
        )

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 practical Python skills for everyday tasks, from calculations and data organization to visualizing information and automating routines. This course blends engaging theory with hands-on challenges, empowering you to solve real-world problems efficiently using Python.

Learn to use Python for automating calculations and streamlining daily routines, from budgeting to simple reminders.

Discover how Python can help you organize, analyze, and summarize everyday data, from to-do lists to simple statistics.

Learn to create simple visualizations and formatted outputs to communicate information clearly and effectively.

Challenge: Visualize Your Weekly Spending

Solution

Challenge: Visualize Your Weekly Spending

Solution