test: cover CP HiCache host integration

2026-05-08 02:59:27 +08:00
parent 39f40c02b4
commit 3e9152194b
4 changed files with 41 additions and 15 deletions
@@ -19,6 +19,24 @@ from sglang.test.test_utils import CustomTestCase
 register_cpu_ci(est_time=2, suite="stage-a-test-cpu")


+class TestCpHiCacheImports(CustomTestCase):
+    def test_cp_hicache_public_imports_without_sgl_kernel(self):
+        import subprocess
+
+        subprocess.run(
+            [
+                sys.executable,
+                "-c",
+                "from sglang.srt.mem_cache.hiradix_cache import HiRadixCache, CpHiCacheNodeMetadata; "
+                "from sglang.srt.managers.cache_controller import HiCacheController; "
+                "print('OK')",
+            ],
+            check=True,
+            capture_output=True,
+            text=True,
+        )
+
+
 class TestCpHiCacheNodeMetadata(CustomTestCase):
    def test_split_zero_len_moves_all_positions_to_child(self):
        metadata = CpHiCacheNodeMetadata(