[Test] enhance test_4d_conversion with opset=11,13 Resize variants

Author: maltanar

tests/transformation/test_4d_conversion.py

@@ -31,6 +31,7 @@
 import numpy as np
 import onnx
 import onnx.parser as oprs
+from onnx.helper import make_opsetid
 
 import qonnx.core.onnx_exec as oxe
 from qonnx.core.datatype import DataType
@@ -328,7 +329,7 @@ def create_conv_upsample():
     return model
 
 
-def create_resize():
+def create_resize(opset):
     """
     Creates an model for testing the 3D to 4D transform of the resize node.
     """
@@ -380,16 +381,24 @@ def create_resize():
         value_info=list_of_value_infos,
     )
 
-    onnx_model = qonnx_make_model(graph, producer_name="4d_conversion_resize_test-model")
+    onnx_model = qonnx_make_model(
+        graph, producer_name="4d_conversion_resize_test-model", opset_imports=[make_opsetid("", opset)]
+    )
     model = ModelWrapper(onnx_model)
-    model = model.transform(InferShapes())
+
     model.set_initializer("sizes_resize1", np.array([1, 32, 8], dtype=np.int64))
-    model.set_initializer("scales_resize1", np.array([], dtype=np.float32))
+    if opset == 11:
+        model.set_initializer("scales_resize1", np.array([], dtype=np.float32))
+    elif opset == 13:
+        model.graph.node[0].input[2] = ""
+    else:
+        assert False, f"Undefined opset {opset} for Resize testcase creator"
     model.set_initializer("scales_resize2", np.array([1.0, 1.0, 2.0], dtype=np.float32))
+    model = model.transform(InferShapes())
     return model
 
 
-@pytest.mark.parametrize("test_model", ["Quartz", "VGG", "ConvUpsample", "Resize"])
+@pytest.mark.parametrize("test_model", ["Quartz", "VGG", "ConvUpsample", "Resize11", "Resize13"])
 def test_4d_conversion(test_model):
     """
     Test for the 3D to 4D transformation with a valid graph.
@@ -401,8 +410,8 @@ def test_4d_conversion(test_model):
         model = create_arbitrary_model_vgg()
     elif test_model == "ConvUpsample":
         model = create_conv_upsample()
-    elif test_model == "Resize":
-        model = create_resize()
+    elif "Resize" in test_model:
+        model = create_resize(opset=int(test_model.replace("Resize", "")))
     else:
         raise Exception("Unknown test_model in test_4d_conversion")