code: def test_create_pyramid_input_masks(self):
      actual = model_utils.create_pyramid_input_masks(
          input_mask=np.ones((2, 128 + 64 + 32), dtype=np.int32),
          num_features_level0=128,
          num_pyramid_levels=3,
          feature_pyramid_downsample_stride=2,
          num_text_tokens=0,
      )
      self.assertLen(actual, 3)
      np.testing.assert_equal(
          actual[0], np.ones((2, 128), dtype=np.int32)
      )
      np.testing.assert_equal(
          actual[1], np.ones((2, 64), dtype=np.int32)
      )
      np.testing.assert_equal(
          actual[2], np.ones((2, 32), dtype=np.int32)
      )
  
file_path: scenic/scenic/projects/unloc/model_utils_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax
  from flax import struct
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.projects.unloc import model_utils
  from scenic.train_lib import train_utils
label_desc: Use image node_id 4 for calling the ModelUtilsTest obj's underlying member method code with example usage: obj.test_create_pyramid_input_masks() without return types