39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
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.metrics.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()
|