code: def test_model_main_input_name(self):
      for model_class in self.all_model_classes:
          model_signature = inspect.signature(
              getattr(model_class, "forward")
          )
          # The main input is the name of the argument after `self`
          observed_main_input_name_nodes = list(
              model_signature.parameters.keys()
          )[1]
          observed_main_input_name_edges = list(
              model_signature.parameters.keys()
          )[2]
          self.assertEqual(
              model_class.main_input_name_nodes,
              observed_main_input_name_nodes,
          )
          self.assertEqual(
              model_class.main_input_name_edges,
              observed_main_input_name_edges,
          )
  
imports: import copy
  import inspect
  import os
  import tempfile
  import unittest
  from transformers import GraphormerConfig, is_torch_available
  from transformers.testing_utils import require_torch, slow, torch_device
  from ...test_configuration_common import ConfigTester
  from ...test_modeling_common import ModelTesterMixin, _config_zero_init, ids_tensor
  from ...test_pipeline_mixin import PipelineTesterMixin
label_desc: Use image node_id 10 for calling the GraphormerModelTest obj's underlying member method code with example usage: obj.test_model_main_input_name() without return types