code: def test_merge_pyramid_input_masks_with_text_mask(self):
      actual = model_utils.merge_pyramid_input_masks(
          [
              np.ones((2, 128), dtype=np.int32),
              np.ones((2, 64), dtype=np.int32),
          ],
          input_text_mask=np.ones((2, 8), dtype=np.int32),
      )
      np.testing.assert_equal(
          actual, np.ones((2, 128 + 8 + 64 + 8), 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 6 for calling the ModelUtilsTest obj's underlying member method code with example usage: obj.test_merge_pyramid_input_masks_with_text_mask() without return types