21 lines
778 B
Python
21 lines
778 B
Python
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
||
|
|
import copy
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from vllm_omni.engine.request_snapshot import copy_request_snapshot
|
||
|
|
|
||
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
||
|
|
|
||
|
|
|
||
|
|
def test_full_snapshot_preserves_fields_aliases_and_isolation():
|
||
|
|
samples = [0.25, 0.5]
|
||
|
|
source = {"prompt_token_ids": [1, 2], "additional_information": {"samples": samples}, "alias": samples}
|
||
|
|
result = copy_request_snapshot(source)
|
||
|
|
assert result == copy.deepcopy(source)
|
||
|
|
assert result["alias"] is result["additional_information"]["samples"]
|
||
|
|
result["alias"].append(0.75)
|
||
|
|
result["prompt_token_ids"][0] = 9
|
||
|
|
assert samples == [0.25, 0.5]
|
||
|
|
assert source["prompt_token_ids"] == [1, 2]
|