code: def test_infer_shape(self):
      A = self.A
      S = self.S
      self._compile_and_check(
          [A],  # pytensor.function inputs
          self.op(A),  # pytensor.function outputs
          # S must be square
          [S],
          self.op_class,
          warn=False,
      )
  
imports: import numpy
  import numpy.linalg
  import pytest
  from numpy import inf
  from numpy.testing import assert_array_almost_equal
  import pytensor
  from pytensor import function
  from pytensor.configdefaults import config
  from pytensor.tensor.math import _allclose
  from pytensor.tensor.nlinalg import SVD, Eig, MatrixInverse, TensorInv, det, eig, eigh, lstsq, matrix_dot, matrix_inverse, matrix_power, norm, pinv, qr, slogdet, svd, tensorinv, tensorsolve, trace
  from pytensor.tensor.type import lmatrix, lscalar, matrix, scalar, tensor3, tensor4, vector
  from tests import unittest_tools
label_desc: Use image node_id 2 for calling the TestEig obj's underlying member method code with example usage: obj.test_infer_shape() without return types