routers/ai/tests/test_shapes.py

from unittest import TestCase
from src.shapes import Circle, Rectangle

class TestShapes(TestCase):
    def test_circle_area(self):
        circle = Circle(radius=5)
        self.assertAlmostEqual(circle.area(), 78.5)

    def test_rectangle_area(self):
        rectangle = Rectangle(width=4, height=5)
        self.assertEqual(rectangle.area(), 20)