[Test] Enhance radix cache test for spec cases (#12394)

This commit is contained in:
Liangsheng Yin
2025-10-30 18:46:28 +08:00
committed by GitHub
parent df5192cffa
commit 73dfd2dfb1
6 changed files with 61 additions and 56 deletions
+5 -1
View File
@@ -4,7 +4,8 @@ from types import SimpleNamespace
from sglang.srt.environ import envs
from sglang.srt.utils import kill_process_tree
from sglang.test.few_shot_gsm8k import run_eval
from sglang.test.kit_matched_stop import MatchedStopMixin
from sglang.test.kits.matched_stop_kit import MatchedStopMixin
from sglang.test.kits.radix_cache_server_kit import run_radix_attention_test
from sglang.test.test_utils import (
DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
DEFAULT_EAGLE_TARGET_MODEL_FOR_TEST,
@@ -65,6 +66,9 @@ class TestEagleServerBase(CustomTestCase, MatchedStopMixin):
def tearDownClass(cls):
kill_process_tree(cls.process.pid)
def test_radix_attention(self):
run_radix_attention_test(self.base_url)
def test_gsm8k(self):
args = SimpleNamespace(
num_shots=5,