From 91a818629c09db1764069fee9089c196fc69515d Mon Sep 17 00:00:00 2001 From: Yueqian Lin <70319226+linyueqian@users.noreply.github.com> Date: Wed, 25 Mar 2026 04:26:38 -0400 Subject: [PATCH] [Cleanup] Remove stray test file from engine directory (#2161) Signed-off-by: linyueqian Co-authored-by: Canlin Guo --- vllm_omni/engine/test_serialization.py | 35 -------------------------- 1 file changed, 35 deletions(-) delete mode 100644 vllm_omni/engine/test_serialization.py diff --git a/vllm_omni/engine/test_serialization.py b/vllm_omni/engine/test_serialization.py deleted file mode 100644 index 69c7fb8a1..000000000 --- a/vllm_omni/engine/test_serialization.py +++ /dev/null @@ -1,35 +0,0 @@ -import torch - -from vllm_omni.engine.serialization import deserialize_additional_information, serialize_additional_information - - -def test_serialize_additional_information(): - info = { - "tensor_f32": torch.rand(3, dtype=torch.float32), - "tensor_f64": torch.rand(3, 3, dtype=torch.float64), - "list": [1, 2, 3], - "scalar": 1, - } - - payload = serialize_additional_information(info) - - assert payload.entries["tensor_f32"].tensor_dtype == "float32" - assert payload.entries["tensor_f32"].tensor_shape == [3] - assert payload.entries["tensor_f32"].tensor_data == info["tensor_f32"].numpy().tobytes() - - assert payload.entries["tensor_f64"].tensor_dtype == "float64" - assert payload.entries["tensor_f64"].tensor_shape == [3, 3] - assert payload.entries["tensor_f64"].tensor_data == info["tensor_f64"].numpy().tobytes() - - assert payload.entries["list"].list_data == info["list"] - assert payload.entries["scalar"].scalar_data == info["scalar"] - - deserialized = deserialize_additional_information(payload) - - # compare tensors separately first - assert torch.equal(deserialized["tensor_f32"], info["tensor_f32"]) - assert torch.equal(deserialized["tensor_f64"], info["tensor_f64"]) - del deserialized["tensor_f32"], info["tensor_f32"] - del deserialized["tensor_f64"], info["tensor_f64"] - - assert deserialized == info