code: def _get_test_model():
      class TestModel(torch.nn.Module):
          def __init__(self):
              super().__init__()
              self.conv1 = torch.nn.Conv2d(1, 8, 5, padding="same")
              self.bn1 = torch.nn.BatchNorm2d(8)
              self.linear1 = torch.nn.Linear(28 * 28 * 8, 5)
              self.alpha = 0.7
  
          def forward(self, x):
              x = self.conv1(x)
              x = self.bn1(x)
              x = self.linear1(torch.flatten(x))
              x = torch.maximum(self.alpha * x, x)
              return x
  
      return TestModel().eval()
  
imports: import copy
  import os
  import tempfile
  import numpy
  import pytest
  import coremltools
  from coremltools._deps import _HAS_TORCH
  from coremltools.converters.mil import Builder
  from coremltools.converters.mil.mil import Function, Program, get_new_symbol
  from coremltools.converters.mil.testing_utils import get_op_types_in_program
label_desc: Use image node_id 1 for calling the TestGraphPassManagement obj's underlying member method code with example usage: obj._get_test_model() and returns: TestModel, x