add uts for multistream

Author: SunnyLee151064

tests/ut/multistream/test_metadata.py

@@ -3,9 +3,9 @@
 import torch
 from tests.ut.base import TestBase 
 from vllm_ascend.multistream.base import MSEventKey
-from vllm_ascend.multistream.metadata import (MultiStreamStepMetadata, 
-                                              MultiStreamConfig,
+from vllm_ascend.multistream.metadata import (MultiStreamConfig,
                                               MultiStreamMetadata,
+                                              MultiStreamStepMetadata, 
                                               split_micro_batches_tensors) 
 
 

tests/ut/multistream/test_ms_split.py

@@ -4,10 +4,10 @@
 import torch 
 from vllm_ascend.attention.attention_v1 import AscendAttentionState
 from vllm_ascend.multistream.base import MSAttentionMetadataSplitConfig
-from vllm_ascend.multistream.ms_split import (compute_split_seq_index, 
+from vllm_ascend.multistream.ms_split import (compute_split_seq_index,
+                                              model_input_split_v1_mla_attn,
                                               split_attn_tensor_type, 
-                                              split_attn_int_type, 
-                                              model_input_split_v1_mla_attn)
+                                              split_attn_int_type)
 
 
 class TestMsSplit(unittest.TestCase):