Update tests/test_torchvision_nodes.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2025-06-21 21:19:59 -06:00
parent a26ecd410e
commit fe60b14624
1 changed files with 2 additions and 0 deletions
@@ -23,7 +23,9 @@ def test_TorchVisionTransformStringParser():
    transforms, name = ret
    assert name == "test"
    assert transforms[0][0] == "Resize"
+    assert transforms[0][1] == ["64", "64"]
    assert transforms[1][0] == "GaussianBlur"
+    assert transforms[1][1] == ["1"]

 def test_TorchVisionTransformNode():
    node1 = TorchVisionTransformCompositionList()