routers/ai/tests/test_math_operations.py

from unittest import TestCase
from src.math_operations import add, subtract

class TestMathOperations(TestCase):
    def test_add(self):
        self.assertEqual(add(2, 3), 5)

    def test_subtract(self):
        self.assertEqual(subtract(5, 3), 2)