code: def test_extract_pyramid_video_tokens(
      self, num_pyramid_levels, input_tokens, expected_output
  ):
      actual = model_utils.extract_pyramid_video_tokens(
          input_tokens,
          num_pyramid_levels=num_pyramid_levels,
          feature_pyramid_downsample_stride=2,
          num_video_tokens_level0=8,
          num_text_tokens=1,
      )
      np.testing.assert_equal(actual, expected_output)
  
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 2 for calling the ModelUtilsTest obj's underlying member method code with example usage: obj.test_extract_pyramid_video_tokens(num_pyramid_levels, input_tokens, expected_output) without return types