code: def test_save_model_picks_correct_format_from_extension(self) -> None:
      proto = _simple_model()
      with tempfile.TemporaryDirectory() as temp_dir:
          model_path = os.path.join(temp_dir, "model.textproto")
          # No format is specified, so the extension should be used to determine the format
          onnx.save_model(proto, model_path)
          loaded_proto = onnx.load_model(model_path, format="textproto")
          self.assertEqual(proto, loaded_proto)
  
imports: import io
  import os
  import pathlib
  import tempfile
  import unittest
  import google.protobuf.message
  import google.protobuf.text_format
  import parameterized
  import onnx
  from onnx import serialization
label_desc: Use image node_id 1 for calling the TestSaveAndLoadFileExtensions obj's underlying member method code with example usage: obj.test_save_model_picks_correct_format_from_extension() without return types