From f27c6cdc2b6a9fe9a6c7da172f283a0772773d4f Mon Sep 17 00:00:00 2001
From: Alison Shao <54658187+alisonshao@users.noreply.github.com>
Date: Mon, 5 Jan 2026 22:21:49 -0800
Subject: [PATCH] Migrate metrics tests to test/registered/metrics/ (#16466)

---
 test/{srt => registered/metrics}/test_metrics.py       | 5 +++++
 test/{srt => registered/metrics}/test_metrics_utils.py | 3 +++
 test/srt/run_suite.py                                  | 4 ----
 3 files changed, 8 insertions(+), 4 deletions(-)
 rename test/{srt => registered/metrics}/test_metrics.py (96%)
 rename test/{srt => registered/metrics}/test_metrics_utils.py (97%)

diff --git a/test/srt/test_metrics.py b/test/registered/metrics/test_metrics.py
similarity index 96%
rename from test/srt/test_metrics.py
rename to test/registered/metrics/test_metrics.py
index 373318201..b429247e8 100644
--- a/test/srt/test_metrics.py
+++ b/test/registered/metrics/test_metrics.py
@@ -2,6 +2,11 @@ import unittest
 from typing import Dict, List
 
 import requests
+
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
+
+register_cuda_ci(est_time=32, suite="stage-b-test-small-1-gpu")
+register_amd_ci(est_time=32, suite="stage-b-test-small-1-gpu")
 from prometheus_client.parser import text_string_to_metric_families
 from prometheus_client.samples import Sample
 
diff --git a/test/srt/test_metrics_utils.py b/test/registered/metrics/test_metrics_utils.py
similarity index 97%
rename from test/srt/test_metrics_utils.py
rename to test/registered/metrics/test_metrics_utils.py
index 1a93a75e0..b88656626 100644
--- a/test/srt/test_metrics_utils.py
+++ b/test/registered/metrics/test_metrics_utils.py
@@ -1,6 +1,9 @@
 import unittest
 
 from sglang.srt.metrics.utils import generate_buckets, two_sides_exponential_buckets
+from sglang.test.ci.ci_register import register_cpu_ci
+
+register_cpu_ci(est_time=1, suite="stage-a-cpu-only")
 
 
 class TestMetricsUtils(unittest.TestCase):
diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index 381eaef07..f293d0144 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -44,8 +44,6 @@ suites = {
         TestFile("test_io_struct.py", 8),
         TestFile("test_jinja_template_utils.py", 7),
         TestFile("test_mamba_unittest.py", 9),
-        TestFile("test_metrics.py", 32),
-        TestFile("test_metrics_utils.py", 1),
         TestFile("test_model_hooks.py", 6),
         TestFile("test_modelopt_loader.py", 11),
         TestFile("test_multi_tokenizer.py", 230),
@@ -190,8 +188,6 @@ suite_amd = {
         TestFile("test_input_embeddings.py", 38),
         TestFile("test_io_struct.py", 8),
         TestFile("test_jinja_template_utils.py", 1),
-        TestFile("test_metrics.py", 32),
-        TestFile("test_metrics_utils.py", 1),
         TestFile("test_model_hooks.py", 10),
         TestFile("test_multi_tokenizer.py", 345),
         TestFile("test_no_chunked_prefill.py", 108),