Update test

Author: justinchuby

src/onnx_ir/serde_test.py

@@ -80,29 +80,53 @@ def test_from_to_onnx_text_with_initializers(self):
    opset_import: ["" : 17]
 >
 agraph (float[1] input_x, float[2] input_y) => (float[2] result) {
-   [node_name] result = Add (input_x, input_y)
+   [node_1] add = Add (input_x, input_y)
+   [node_2] result = Add (add, initializer_z)
 }"""
         self.maxDiff = None
         array = np.array([1.0, 2.0], dtype=np.float32)
+        init_array = np.array([3.0, 4.0], dtype=np.float32)
         model = serde.from_onnx_text(
-            model_text, initializers=[ir.tensor(array, name="input_y")]
+            model_text,
+            initializers=[
+                ir.tensor(init_array, name="initializer_z"),
+                ir.tensor(array, name="input_y"),
+            ],
         )
         np.testing.assert_array_equal(model.graph.inputs[1].const_value.numpy(), array)
+        np.testing.assert_array_equal(
+            model.graph.initializers["initializer_z"].const_value.numpy(), init_array
+        )
         expected_text = """\
 <
    ir_version: 10,
    opset_import: ["" : 17]
 >
 agraph (float[1] input_x, float[2] input_y) => (float[2] result)
-   <float[2] input_y =  {1,2}>
+   <float[2] initializer_z =  {3,4}, float[2] input_y =  {1,2}>
 {
-   [node_name] result = Add (input_x, input_y)
+   [node_1] add = Add (input_x, input_y)
+   [node_2] result = Add (add, initializer_z)
 }"""
         onnx_text_roundtrip = serde.to_onnx_text(model)
         stripped_lines = [line.rstrip() for line in onnx_text_roundtrip.splitlines()]
         result = "\n".join(stripped_lines)
         self.assertEqual(result, expected_text)
 
+    def test_to_onnx_text_excluding_initializers(self):
+        model_text = """\
+<
+   ir_version: 10,
+   opset_import: ["" : 17]
+>
+agraph (float[1] input_x, float[2] input_y) => (float[2] result) {
+   [node_name] result = Add (input_x, input_y)
+}"""
+        self.maxDiff = None
+        array = np.array([1.0, 2.0], dtype=np.float32)
+        model = serde.from_onnx_text(
+            model_text, initializers=[ir.tensor(array, name="input_y")]
+        )
         onnx_text_without_initializers = serde.to_onnx_text(model, exclude_initializers=True)
         expected_text_without_initializers = """\
 <