[CPU] use CustomTestCase for registered CPU tests (#38244)

2026-09-07 15:09:52 +08:00
parent a88e852fab
commit 6a1ff90f2d
4 changed files with 8 additions and 5 deletions
@@ -6,6 +6,7 @@ from starlette.datastructures import Headers

 from sglang.srt.entrypoints.request_headers import apply_header_overrides
 from sglang.test.ci.ci_register import register_cpu_ci
+from sglang.test.test_utils import CustomTestCase

 register_cpu_ci(est_time=4, suite="stage-a-test-cpu-intel")

@@ -23,7 +24,7 @@ def _obj():
    )


-class TestApplyRoutingHeaders(unittest.TestCase):
+class TestApplyRoutingHeaders(CustomTestCase):
    def test_sets_all_fields_with_types(self):
        obj = _obj()
        apply_header_overrides(