
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

Вивчайте основи геометричного моделювання за допомогою Python. Досліджуйте способи представлення, маніпулювання та апроксимації геометричних фігур програмними методами. Курс охоплює базові форми, перетворення та практичні методи апроксимації, роблячи геометричні концепції доступними та інтерактивними.

Ознайомлення з основними концепціями геометричного моделювання та використанням Python для представлення і маніпулювання базовими фігурами.

Занурення у геометричні перетворення, такі як перенесення, обертання та масштабування, і аналіз їх впливу на фігури у Python.

Дізнайтеся, як апроксимувати криві та складні форми за допомогою багатокутників та інших методів у Python.

Challenge: Visual Comparison of Approximations

Рішення