code: def test_shift_right(self):
      # Test shifts right on axis=1
      layer = tl.ShiftRight()
      x = np.array(
          [
              [[9, 9, 9], [8, 8, 8], [7, 7, 7], [6, 6, 6]],
              [[99, 98, 97], [96, 95, 94], [93, 92, 91], [90, 89, 88]],
          ]
      )
      y = layer(x)
      self.assertEqual(x.shape, y.shape)
      self.assertEqual(
          tl.to_list(y),
          [
              [[0, 0, 0], [9, 9, 9], [8, 8, 8], [7, 7, 7]],
              [[0, 0, 0], [99, 98, 97], [96, 95, 94], [93, 92, 91]],
          ],
      )
  
imports: import functools
  from absl.testing import absltest
  import numpy
  from trax import shapes
  import trax.layers
  from trax.layers import test_utils
label_desc: Use image node_id 2 for calling the AttentionTest obj's underlying member method code with example usage: obj.test_shift_right() without return types