test_servant.py

from patterns.behavioral.servant import GeometryTools, Circle, Rectangle, Position
import pytest
import math


@pytest.fixture
def circle():
    return Circle(3, Position(0, 0))

@pytest.fixture
def rectangle():
    return Rectangle(4, 5, Position(0, 0))


def test_calculate_area(circle, rectangle):
    assert GeometryTools.calculate_area(circle) == math.pi * 3 ** 2
    assert GeometryTools.calculate_area(rectangle) == 4 * 5

    with pytest.raises(ValueError):
        GeometryTools.calculate_area("invalid shape")

def test_calculate_perimeter(circle, rectangle):
    assert GeometryTools.calculate_perimeter(circle) == 2 * math.pi * 3
    assert GeometryTools.calculate_perimeter(rectangle) == 2 * (4 + 5)

    with pytest.raises(ValueError):
        GeometryTools.calculate_perimeter("invalid shape")


def test_move_to(circle, rectangle):
    new_position = Position(1, 1)
    GeometryTools.move_to(circle, new_position)
    assert circle.position == new_position

    new_position = Position(1, 1)
    GeometryTools.move_to(rectangle, new_position)
    assert rectangle.position == new_position