[CI] Move existing unit tests into unit directory (#20631)
This commit is contained in:
@@ -0,0 +1,38 @@
|
||||
import time
|
||||
import unittest
|
||||
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=60, suite="default", nightly=True)
|
||||
|
||||
|
||||
class TestCpuMonitor(unittest.TestCase):
|
||||
def test_cpu_monitor(self):
|
||||
from prometheus_client import REGISTRY
|
||||
|
||||
from sglang.srt.observability.cpu_monitor import start_cpu_monitor_thread
|
||||
|
||||
thread = start_cpu_monitor_thread("test", interval=0.1)
|
||||
self.assertTrue(thread.is_alive())
|
||||
self.assertTrue(thread.daemon)
|
||||
|
||||
end_time = time.monotonic() + 0.3
|
||||
while time.monotonic() < end_time:
|
||||
_ = sum(i * i for i in range(1000))
|
||||
time.sleep(0.2)
|
||||
|
||||
value = None
|
||||
for metric in REGISTRY.collect():
|
||||
for sample in metric.samples:
|
||||
if (
|
||||
sample.name == "sglang:process_cpu_seconds_total"
|
||||
and sample.labels.get("component") == "test"
|
||||
):
|
||||
value = sample.value
|
||||
print(f"sglang:process_cpu_seconds_total = {value}")
|
||||
self.assertIsNotNone(value)
|
||||
self.assertGreater(value, 0)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user