From c42fe0b63a29d3ec157089c9784643000dde4aec Mon Sep 17 00:00:00 2001 From: TeeKen Lau <13831887+teekenl@users.noreply.github.com> Date: Wed, 13 Aug 2025 02:34:41 +1000 Subject: [PATCH] Add more test scenario for tensor schema (#22733) Signed-off-by: teekenl --- tests/utils_/test_tensor_schema.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/utils_/test_tensor_schema.py b/tests/utils_/test_tensor_schema.py index 69744921b16a8..6aa781c1564de 100644 --- a/tests/utils_/test_tensor_schema.py +++ b/tests/utils_/test_tensor_schema.py @@ -33,6 +33,31 @@ def test_tensor_schema_constant_dim_failure(): ) +def test_tensor_schema_invalid_types_in_list(): + with pytest.raises(ValueError, match="is not a torch.Tensor"): + Phi3VImagePixelInputs( + data=[ + torch.randn(64, 3, 32, 32), + "not_a_tensor", + torch.randn(64, 3, 32, 32), + ], + image_sizes=torch.randint(0, 256, (3, 2)), + ) + + +def test_tensor_schema_rank_mismatch(): + with pytest.raises(ValueError, match="has rank 3 but expected 5"): + Phi3VImagePixelInputs( + data=torch.randn(16, 64, 3), + image_sizes=torch.randint(0, 256, (16, 2)), + ) + + +def test_tensor_schema_missing_required_field(): + with pytest.raises(ValueError, match="Required field 'data' is missing"): + Phi3VImagePixelInputs(image_sizes=torch.randint(0, 256, (16, 2)), ) + + def test_tensor_schema_symbolic_dim_mismatch(): with pytest.raises(ValueError, match="expected 'bn'=12, got 16"): Phi3VImagePixelInputs(