import os import sys # Ensure the local src/ package is importable for the test runner ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir)) SRC = os.path.join(ROOT, 'src') if SRC not in sys.path: sys.path.insert(0, SRC) from catopt_category_theoretic_compositional.__init__ import add def test_add_basic(): assert add(2, 3) == 5