code: def test_shape(self, input_param, expected_shape):
      net = Transchex(**input_param)
      with eval_mode(net):
          result = net(
              torch.randint(2, (2, 512)),
              torch.randint(2, (2, 512)),
              torch.randn((2, 3, 224, 224)),
          )
          self.assertEqual(result.shape, expected_shape)
  
file_path: MONAI/tests/test_transchex.py
imports: from __future__ import annotations
  import unittest
  import torch
  from parameterized import parameterized
  from monai.networks import eval_mode
  from monai.networks.nets.transchex import Transchex
  from tests.utils import skip_if_quick
label_desc: Use image node_id 1 for calling the TestTranschex obj's underlying member method code with example usage: obj.test_shape(input_param, expected_shape) without return types