From a75840b37339869c8ec7edafdf952a4f616bf573 Mon Sep 17 00:00:00 2001
From: Mick <mickjagger19@icloud.com>
Date: Sun, 1 Mar 2026 19:38:20 +0800
Subject: [PATCH] [diffusion] CI: create and refactor UT (#19619)

---
 python/sglang/multimodal_gen/test/run_suite.py         | 10 +++++++---
 .../test/{server => unit}/test_lora_format_adapter.py  |  0
 .../test/{ => unit}/test_sampling_params_validate.py   |  0
 .../multimodal_gen/test/{ => unit}/test_storage.py     |  0
 4 files changed, 7 insertions(+), 3 deletions(-)
 rename python/sglang/multimodal_gen/test/{server => unit}/test_lora_format_adapter.py (100%)
 rename python/sglang/multimodal_gen/test/{ => unit}/test_sampling_params_validate.py (100%)
 rename python/sglang/multimodal_gen/test/{ => unit}/test_storage.py (100%)

diff --git a/python/sglang/multimodal_gen/test/run_suite.py b/python/sglang/multimodal_gen/test/run_suite.py
index fc5224774..046eb6005 100644
--- a/python/sglang/multimodal_gen/test/run_suite.py
+++ b/python/sglang/multimodal_gen/test/run_suite.py
@@ -29,14 +29,18 @@ _UPDATE_WEIGHTS_MODEL_PAIR_IDS = (
 )
 
 SUITES = {
+    # no GPU required; safe to run on any CPU-only runner
+    "unit": [
+        "../unit/test_sampling_params_validate.py",
+        "../unit/test_storage.py",
+        "../unit/test_lora_format_adapter.py",
+        # add new unit tests here
+    ],
     "1-gpu": [
         "test_server_a.py",
         "test_server_b.py",
-        "test_lora_format_adapter.py",
         # cli test
         "../cli/test_generate_t2i_perf.py",
-        # unit tests (no server needed)
-        "../test_sampling_params_validate.py",
         "test_update_weights_from_disk.py",
         # add new 1-gpu test files here
     ],
diff --git a/python/sglang/multimodal_gen/test/server/test_lora_format_adapter.py b/python/sglang/multimodal_gen/test/unit/test_lora_format_adapter.py
similarity index 100%
rename from python/sglang/multimodal_gen/test/server/test_lora_format_adapter.py
rename to python/sglang/multimodal_gen/test/unit/test_lora_format_adapter.py
diff --git a/python/sglang/multimodal_gen/test/test_sampling_params_validate.py b/python/sglang/multimodal_gen/test/unit/test_sampling_params_validate.py
similarity index 100%
rename from python/sglang/multimodal_gen/test/test_sampling_params_validate.py
rename to python/sglang/multimodal_gen/test/unit/test_sampling_params_validate.py
diff --git a/python/sglang/multimodal_gen/test/test_storage.py b/python/sglang/multimodal_gen/test/unit/test_storage.py
similarity index 100%
rename from python/sglang/multimodal_gen/test/test_storage.py
rename to python/sglang/multimodal_gen/test/unit/test_storage.py