test: fix Task 1 CPU unit coverage

This commit is contained in:
2026-05-08 00:29:33 +08:00
parent 95bacb8862
commit 36ee803aa7
2 changed files with 14 additions and 3 deletions
@@ -17,6 +17,13 @@ Usage:
python -m pytest test_radix_cache_unit.py::TestRadixCache::test_insert_basic
"""
import sys
import unittest.mock
for _mod in ("sgl_kernel", "sgl_kernel.kvcacheio"):
if _mod not in sys.modules:
sys.modules[_mod] = unittest.mock.MagicMock()
from sglang.srt.mem_cache.common import available_and_evictable_str
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
@@ -27,7 +34,6 @@ register_amd_ci(est_time=5, suite="stage-b-test-1-gpu-small-amd")
import random
import time
import unittest
import unittest.mock
import torch
@@ -743,6 +749,7 @@ class TestRadixCache(unittest.TestCase):
# Should have 1 page (split at page_size=2)
self.assertEqual(len(node.hash_value), 1)
@unittest.skipIf(not torch.cuda.is_available(), "CUDA is required")
def test_memory_allocated(self):
keys, values = [], []