fix unit test

xadupre · xadupre · commit 0c2a92d5ef65 · 2024-04-01T10:57:35.000+02:00
diff --git a/_unittests/ut_translate_api/test_translate.py b/_unittests/ut_translate_api/test_translate.py
@@ -80,9 +80,9 @@ def test_transpose(self):
             """
             (
                 start(opset=19)
+                .vin('X', elem_type=TensorProto.FLOAT)
                 .cst(np.array([-1, 1], dtype=np.int64))
                 .rename('r')
-                .vin('X', elem_type=TensorProto.FLOAT)
                 .bring('X', 'r')
                 .Reshape()
                 .rename('r0_0')
@@ -166,9 +166,9 @@ def test_export_if(self):
             f"""
             (
                 start(opset=19)
+                .vin('X', elem_type=TensorProto.FLOAT)
                 .cst(np.array([0.0], dtype=np.float32))
                 .rename('r')
-                .vin('X', elem_type=TensorProto.FLOAT)
                 .bring('X')
                 .ReduceSum(keepdims=1, noop_with_empty_axes=0)
                 .rename('Xs')
@@ -202,9 +202,9 @@ def test_aionnxml(self):
             """
             (
                 start(opset=19, opsets={'ai.onnx.ml': 3})
+                .vin('X', elem_type=TensorProto.FLOAT)
                 .cst(np.array([-1, 1], dtype=np.int64))
                 .rename('r')
-                .vin('X', elem_type=TensorProto.FLOAT)
                 .bring('X', 'r')
                 .Reshape()
                 .rename('USE')
diff --git a/_unittests/ut_translate_api/test_translate_classic.py b/_unittests/ut_translate_api/test_translate_classic.py
@@ -138,13 +138,13 @@ def test_transpose(self):
             initializers = []
             sparse_initializers = []
             functions = []
+            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             initializers.append(
                 from_array(
                     np.array([-1, 1], dtype=np.int64),
                     name='r'
                 )
             )
-            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             nodes.append(
                 make_node_extended(
                     'Reshape',
@@ -278,13 +278,13 @@ def test_aionnxml(self):
             initializers = []
             sparse_initializers = []
             functions = []
+            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             initializers.append(
                 from_array(
                     np.array([-1, 1], dtype=np.int64),
                     name='r'
                 )
             )
-            inputs.append(make_tensor_value_info('X', TensorProto.FLOAT, shape=[]))
             nodes.append(
                 make_node_extended(
                     'Reshape',
diff --git a/onnx_array_api/translate_api/translate.py b/onnx_array_api/translate_api/translate.py
@@ -75,6 +75,8 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:
                     domain=self.proto_.domain,
                 )
             )
+        elif isinstance(self.proto_, GraphProto):
+            rows.extend(self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.name))
         else:
             rows.extend(
                 self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.graph.name)

Original file line number	Diff line number	Diff line change
`@@ -75,6 +75,8 @@ def export(self, as_str, single_line: bool = False) -> Union[str, List[str]]:`
`75`	`75`	`domain=self.proto_.domain,`
`76`	`76`	`)`
`77`	`77`	`)`
	`78`	`+ elif isinstance(self.proto_, GraphProto):`
	`79`	`+ rows.extend(self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.name))`
`78`	`80`	`else:`
`79`	`81`	`rows.extend(`
`80`	`82`	`self.emitter(EventType.BEGIN_GRAPH, name=self.proto_.graph.name)`