packaged test into function fix

tests/test_router_mount.py

@@ -1,20 +1,25 @@
 from fastapi import FastAPI, APIRouter
 from fastapi.testclient import TestClient
 
-app = FastAPI()
-api_router = APIRouter(prefix="/api")
 
-@api_router.get("/app")
+def test_mount_on_router():
-def read_main():
+    app = FastAPI()
-    return {"message": "Hello World from main app"}
+    api_router = APIRouter(prefix="/api")
 
-subapi = FastAPI()
+    @api_router.get("/app")
+    def read_main():
+        return {"message": "Hello World from main app"}
 
-@subapi.get("/sub")
+    subapi = FastAPI()
-def read_sub():
-    return {"message": "Hello World from sub API"}
 
-api_router.mount("/subapi", subapi)  # ← moved up
+    @subapi.get("/sub")
-app.include_router(api_router)       # ← now after
+    def read_sub():
+        return {"message": "Hello World from sub API"}
 
-print("All tests passed.")
+    api_router.mount("/subapi", subapi)
+    app.include_router(api_router)
+
+    client = TestClient(app)
+
+    assert client.get("/api/app").status_code == 200
+    assert client.get("/api/subapi/sub").status_code == 200