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

class TestTask(unittest.TestCase):
    def setUp(self):
        # Patch plt.show to avoid displaying plots during tests
        self._show_backup = plt.show
        plt.show = lambda *args, **kwargs: None

    def tearDown(self):
        plt.show = self._show_backup

    def get_line_data(self, ax):
        # Get all lines from the axes
        lines = ax.get_lines()
        if not lines:
            return [], []
        xdata = lines[0].get_xdata()
        ydata = lines[0].get_ydata()
        return list(xdata), list(ydata)

    def test_connects_vertices_in_order(self):
        import user_code
        importlib.reload(user_code)
        plt.clf()
        vertices = [(0, 0), (2, 0), (1, 2)]
        user_code.plot_polygon(vertices)
        ax = plt.gca()
        xdata, ydata = self.get_line_data(ax)
        expected_x = [0, 2, 1, 0]
        expected_y = [0, 0, 2, 0]
        _dynamic_test(
            self,
            xdata == expected_x and ydata == expected_y,
            "Vertices are connected in the correct order and polygon is closed",
            f"Expected x: {expected_x}, y: {expected_y}; got x: {xdata}, y: {ydata}"
        )

    def test_polygon_closed(self):
        import user_code
        importlib.reload(user_code)
        plt.clf()
        vertices = [(1, 1), (4, 1), (4, 4), (1, 4)]
        user_code.plot_polygon(vertices)
        ax = plt.gca()
        xdata, ydata = self.get_line_data(ax)
        # Check if the last point connects to the first
        _dynamic_test(
            self,
            xdata[0] == xdata[-1] and ydata[0] == ydata[-1],
            "Polygon is closed by connecting last vertex to the first",
            f"Polygon is not closed: start ({xdata[0]}, {ydata[0]}), end ({xdata[-1]}, {ydata[-1]})"
        )

    def test_vertices_as_points(self):
        import user_code
        importlib.reload(user_code)
        plt.clf()
        vertices = [(3, 1), (6, 1), (6, 3), (3, 3)]
        user_code.plot_polygon(vertices)
        ax = plt.gca()
        lines = ax.get_lines()
        _dynamic_test(
            self,
            len(lines) > 0,
            "Plot contains at least one line object",
            "No line object found in plot"
        )
        xdata = lines[0].get_xdata()
        ydata = lines[0].get_ydata()
        # Check that all original vertices are present as points in the plot
        points = set((x, y) for x, y in zip(xdata, ydata))
        for vx, vy in vertices:
            _dynamic_test(
                self,
                (vx, vy) in points,
                f"Vertex ({vx}, {vy}) is plotted as a point",
                f"Vertex ({vx}, {vy}) is missing from the plot"
            )

    def test_handles_three_or_more_vertices(self):
        import user_code
        importlib.reload(user_code)
        plt.clf()
        # Triangle
        vertices3 = [(0, 0), (1, 0), (0.5, 1)]
        user_code.plot_polygon(vertices3)
        ax = plt.gca()
        xdata, ydata = self.get_line_data(ax)
        _dynamic_test(
            self,
            len(xdata) == 4 and len(ydata) == 4,
            "Function handles polygons with three vertices (triangle)",
            f"Triangle not plotted correctly: x={xdata}, y={ydata}"
        )
        plt.clf()
        # Pentagon
        vertices5 = [(0, 0), (1, 0), (1.5, 1), (0.5, 2), (-0.5, 1)]
        user_code.plot_polygon(vertices5)
        ax = plt.gca()
        xdata, ydata = self.get_line_data(ax)
        _dynamic_test(
            self,
            len(xdata) == 6 and len(ydata) == 6,
            "Function handles polygons with five vertices (pentagon)",
            f"Pentagon not plotted correctly: x={xdata}, y={ydata}"
        )

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で多角形やその他の手法を用いて曲線や複雑な形状を近似する方法を学びます。

チャレンジ：ポリゴンプロット

解答