
import unittest
import user_code
import ast
import re   
import importlib
import csv
import unittest
import importlib
from unittest.mock import patch
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import numpy as np

class TestTask(unittest.TestCase):
    def setUp(self):
        import user_code
        importlib.reload(user_code)
        self.user_code = user_code

    def test_circle_subplot_and_legend(self):
        plt.close('all')
        with patch('matplotlib.pyplot.show'):
            self.user_code.plot_curve_approximations()
        figs = [plt.figure(i) for i in plt.get_fignums()]
        _dynamic_test(
            self,
            len(figs) > 0,
            "At least one figure is created.",
            "No figure was created."
        )
        found_subplot = False
        found_legend = False
        for ax in figs[0].axes:
            title = ax.get_title().lower()
            if 'circle' in title:
                found_subplot = True
                handles, labels = ax.get_legend_handles_labels()
                labelset = set([l.lower() for l in labels])
                found_legend = (
                    any('true' in l and 'circle' in l for l in labelset)
                    and any('approx' in l for l in labelset)
                )
        _dynamic_test(
            self,
            found_subplot,
            "Circle subplot found.",
            "No subplot for circle found."
        )
        _dynamic_test(
            self,
            found_legend,
            "Circle subplot contains legend distinguishing true curve and approximation.",
            "Legend for circle subplot missing or incomplete."
        )

    def test_ellipse_subplot_and_legend(self):
        plt.close('all')
        with patch('matplotlib.pyplot.show'):
            self.user_code.plot_curve_approximations()
        figs = [plt.figure(i) for i in plt.get_fignums()]
        _dynamic_test(
            self,
            len(figs) > 0,
            "At least one figure is created.",
            "No figure was created."
        )
        found_subplot = False
        found_legend = False
        for ax in figs[0].axes:
            title = ax.get_title().lower()
            if 'ellipse' in title:
                found_subplot = True
                handles, labels = ax.get_legend_handles_labels()
                labelset = set([l.lower() for l in labels])
                found_legend = (
                    any('true' in l and 'ellipse' in l for l in labelset)
                    and any('approx' in l for l in labelset)
                )
        _dynamic_test(
            self,
            found_subplot,
            "Ellipse subplot found.",
            "No subplot for ellipse found."
        )
        _dynamic_test(
            self,
            found_legend,
            "Ellipse subplot contains legend distinguishing true curve and approximation.",
            "Legend for ellipse subplot missing or incomplete."
        )

    def test_equal_axis_scaling(self):
        plt.close('all')
        with patch('matplotlib.pyplot.show'):
            self.user_code.plot_curve_approximations()
        figs = [plt.figure(i) for i in plt.get_fignums()]
        _dynamic_test(
            self,
            len(figs) > 0,
            "At least one figure is created.",
            "No figure was created."
        )
        for ax in figs[0].axes:
            aspect = ax.get_aspect()
            _dynamic_test(
                self,
                aspect == 'equal' or np.isclose(aspect, 1.0),
                "Axis scaling in subplot is equal.",
                f"Axis scaling is not equal for subplot titled '{ax.get_title()}'."
            )

    def test_polygonal_segments_same_for_both(self):
        plt.close('all')
        # Patch show so plot window does not block
        with patch('matplotlib.pyplot.show'):
            self.user_code.plot_curve_approximations()
        figs = [plt.figure(i) for i in plt.get_fignums()]
        _dynamic_test(
            self,
            len(figs) > 0,
            "At least one figure is created.",
            "No figure was created."
        )
        # Find number of points in polygonal approximations for both subplots
        num_points = []
        for ax in figs[0].axes:
            for line in ax.get_lines():
                label = line.get_label().lower()
                if 'approx' in label:
                    data = line.get_xydata()
                    num_points.append(len(data))
        _dynamic_test(
            self,
            len(num_points) >= 2,
            "Both approximations plotted.",
            "Polygonal approximations for both shapes were not found."
        )
        _dynamic_test(
            self,
            num_points[0] == num_points[1],
            "Polygonal approximations use same number of segments.",
            f"Polygonal approximations have different number of points: {num_points}" 
        )

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

Aprenda os fundamentos da modelagem geométrica utilizando Python. Explore como representar, manipular e aproximar figuras geométricas de forma programática. Este curso aborda formas básicas, transformações e técnicas práticas de aproximação, tornando conceitos geométricos acessíveis e interativos.

Introdução aos conceitos fundamentais de modelagem geométrica e como o Python pode ser utilizado para representar e manipular formas básicas.

Aprofunde-se em transformações geométricas como translação, rotação e escala, e veja como elas afetam formas em Python.

Aprenda a aproximar curvas e formas complexas utilizando polígonos e outras técnicas em Python.

Desafio: Comparação Visual de Aproximações

Solução