code: def test_layer_stack_update_context_position(self):
      x = np.random.randn(2, 3, 4, 5)
      layer_stack, context = create_dummy_model(
          self.converter.mesh, shapes=x.shape
      )
      x_mtf = self.converter.convert_np_array_to_mtf_tensor(
          x, dim_names=self.default_dim_names, dtype=tf.float32
      )
      _ = layer_stack.call(context, x_mtf)
      actual = self.converter.convert_mtf_tensor_to_np_array(
          context.position
      )
      self.assertAllEqual(np.arange(2), actual)
  
file_path: mesh/mesh_tensorflow/transformer/funnel_transformer_test.py
imports: from absl.testing import parameterized
  import mesh_tensorflow
  from mesh_tensorflow import test_utils
  from mesh_tensorflow.transformer import funnel_transformer
  from mesh_tensorflow.transformer import transformer
  from mesh_tensorflow.transformer import transformer_layers
  import numpy
  import tensorflow.compat.v1
label_desc: Use image node_id 7 for calling the FunnelTransformerTest obj's underlying member method code with example usage: obj.test_layer_stack_update_context_position() without return types