diff --git a/tests/routes/__init__.py b/tests/routes/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/agentic_security/routes/test_csv.py b/tests/routes/test_csv.py similarity index 100% rename from agentic_security/routes/test_csv.py rename to tests/routes/test_csv.py diff --git a/agentic_security/routes/test_health.py b/tests/routes/test_health.py similarity index 87% rename from agentic_security/routes/test_health.py rename to tests/routes/test_health.py index 32ce613..09530c4 100644 --- a/agentic_security/routes/test_health.py +++ b/tests/routes/test_health.py @@ -1,6 +1,6 @@ from fastapi.testclient import TestClient -from ..app import app +from agentic_security.app import app def test_health_check(): diff --git a/agentic_security/routes/test_probe.py b/tests/routes/test_probe.py similarity index 97% rename from agentic_security/routes/test_probe.py rename to tests/routes/test_probe.py index 7005976..3525997 100644 --- a/agentic_security/routes/test_probe.py +++ b/tests/routes/test_probe.py @@ -5,10 +5,10 @@ import httpx import pytest from fastapi.testclient import TestClient -from ..app import app -from ..primitives import Probe -from ..probe_actor.refusal import REFUSAL_MARKS -from ..probe_data import REGISTRY +from agentic_security.app import app +from agentic_security.primitives import Probe +from agentic_security.probe_actor.refusal import REFUSAL_MARKS +from agentic_security.probe_data import REGISTRY client = TestClient(app) diff --git a/agentic_security/routes/test_report.py b/tests/routes/test_report.py similarity index 97% rename from agentic_security/routes/test_report.py rename to tests/routes/test_report.py index 237f262..3a887eb 100644 --- a/agentic_security/routes/test_report.py +++ b/tests/routes/test_report.py @@ -4,7 +4,7 @@ from unittest.mock import patch import pytest from fastapi.testclient import TestClient -from .report import router +from agentic_security.routes.report import router client = TestClient(router) diff --git a/agentic_security/routes/test_static.py b/tests/routes/test_static.py similarity index 93% rename from agentic_security/routes/test_static.py rename to tests/routes/test_static.py index d6811d0..f097653 100644 --- a/agentic_security/routes/test_static.py +++ b/tests/routes/test_static.py @@ -4,8 +4,8 @@ import pytest from fastapi import HTTPException from fastapi.testclient import TestClient -from ..primitives import Settings -from .static import get_static_file, router +from agentic_security.primitives import Settings +from agentic_security.routes.static import get_static_file, router client = TestClient(router)