Note: Setup test environment: https://vllm-ascend.readthedocs.io/en/latest/developer_guide/contribution/testing.html

Motivation

This issue attempt to reduce the gap of unit tests to cover the code. There is a brief architecture of ut in tests/ut/ already. We need to add more to cover all the code in vllm-ascend, and there are several principles to follow:

1. The overall file tree should be consistent with vllm_ascend
2. The file name should be the original file name with a prefix test_
3. Use unittest framework, make good use of mock
4. The UTs are all running on cpu node, mock the function related to device to host

Please refer to the official doc on contributing and testing to develop, thanks!

Unit tests need to add

• |-- test__version.py
• |-- test_ascend_config.py
• |-- test_attention
• | |-- test_attention.py will drop later
• | |-- test_attention_v1.py Fix W8A8 fused moe bug #1529
• | `-- test_mla_v1.py @SunnyLee151064
• |-- compilation
• | `-- test_piecewise_backend.py @SunnyLee151064
• |-- core
• | |-- test_schedule_config.py @nuclearwu
• | `-- test_scheduler.py @SunnyLee151064
• |-- device_allocator
• | `-- test_camem.py @1024daniel
• |-- distributed
• | |-- test_communicator.py @FieeFlip
• | |-- test_device_communicators @Agonixiaoxiao add device communitor unit test #1601
• | | |-- test_pyhccl.py @SunnyLee151064
• | | `-- test_pyhccl_wrapper.py @SunnyLee151064
• | |-- kv_transfer @Agonixiaoxiao
• | | |-- test_simple_buffer.py add ut for kv tansfer module #1531
• | | |-- test_simple_connector.py add ut for kv tansfer module #1531
• | | |-- test_simple_pipe.py add ut for kv tansfer module #1531
• | | `-- test_utils.py @cocacolafan
• | |-- test_llmdatadist_connector.py @cocacolafan
• | `-- test_parallel_state.py @wangyanhui-cmss [CI/UT] Add ut for parallel_state.py #1460
• |-- test_envs.py @YuanCheng-coder
• |-- lora
• | `-- punica_wrapper @hongfugui
• | `-- test_punica_npu.py @hongfugui
• |-- models
• | |-- test_deepseek_dbo.py
• | |-- test_deepseek_mtp.py @Liccol
• | |-- test_deepseek_v2.py @ZrBac
• | |-- test_qwen2_5_vl.py @Ronald1995
• | |-- test_qwen2_5_vl_without_padding.py @Ronald1995
• | |-- test_qwen2_vl.py @Ronald1995
• | `-- test_qwen3_moe.py @loukong33
• |-- multistream
• | |-- test_base.py @SunnyLee151064
• | |-- test_context.py @xingLong-xl
• | |-- test_decorator.py @Liccol
• | |-- test_layers.py @1024daniel
• | |-- test_metadata.py @SunnyLee151064
• | `-- test_ms_split.py @SunnyLee151064
• |-- ops
• | |-- test_activation.py @MengqingCao
• | |-- test_attention.py @SunnyLee151064
• | |-- test_cache.py @SunnyLee151064
• | |-- test_common_fused_moe.py @MengqingCao
• | |-- test_expert_load_balancer.py
• | |-- test_fused_moe.py @shiyuan680
• | |-- test_layernorm.py @MengqingCao
• | |-- test_rotary_embedding.py @MengqingCao
• | `-- test_vocab_parallel_embedding.py @YuanCheng-coder
• |-- patch
• | |-- platform
• | | |-- patch_0_9_2
• | | |-- patch_common
• | | | `-- test_test_patch_distributed.py @yangqinghao-cmss
• | | `-- patch_main
• | `-- worker
• | |-- patch_0_9_2
• | |-- patch_common
• | | |-- test_patch_distributed.py @Pr0Wh1teGivee
• | | |-- test_patch_eagle.py @Pr0Wh1teGivee
• | | |-- test_patch_minicpm.py @Pr0Wh1teGivee
• | | |-- test_patch_multi_step_worker.py @Pr0Wh1teGivee
• | | |-- test_patch_sampler.py
• | | |-- test_patch_spec_decode_worker.py @Pr0Wh1teGivee
• | | `-- test_patch_utils.py @Pr0Wh1teGivee
• | `-- patch_main
• |-- test_platform.py @zhanghw0354 [Test]Add unit test for platform.py #1476
• |-- quantization
• | |-- test_func_wrapper.py @xudongLi-cmss
• | |-- test_quant_config.py @nuclearwu Fix W8A8 fused moe bug #1529
• | |-- test_quantizer.py Fix W8A8 fused moe bug #1529
• | |-- test_w8a8.py Fix W8A8 fused moe bug #1529
• | `-- test_w8a8_dynamic.py
• |-- sample
• | |-- ops
• | `-- test_rejection_sampler.py @momo609
• |-- test_utils.py
• `-- worker
• |-- test_cache_engine.py @machenglong2025
• |-- test_draft_model_runner.py @wangyanhui-cmss will drop later
• |-- test_model_runner.py will drop later
• |-- test_model_runner_v1.py @xudongLi-cmss
• |-- test_mtp_proposer_v1.py @machenglong2025
• |-- test_multi_step_runner.py will drop later
• |-- test_multi_step_worker.py will drop later
• |-- test_pooling_model_runner.py @wangyanhui-cmss Add ut for test_pooling_model_runner.py #1640
• |-- test_worker.py @zhanghw0354 will drop later
• `-- test_worker_v1.py @zhanghw0354