code: def test_apply_colormap(self):
      assert image_utils.get_colormap().shape == (256, 3)
  
      gray_img = np.array(
          [
              [[0], [2]],
              [[1234], [2]],
          ],
          dtype=np.uint16,
      )
      assert gray_img.shape == (2, 2, 1)
      colored_img = image_utils.apply_colormap(gray_img)
      assert colored_img.shape == (2, 2, 3)
      assert colored_img.dtype == np.uint8
  
      np.testing.assert_array_equal(
          colored_img,
          [
              [[0, 0, 0], [55, 126, 184]],
              [[71, 84, 183], [55, 126, 184]],
          ],
      )
  
file_path: datasets/tensorflow_datasets/core/utils/image_utils_test.py
imports: import os
  import numpy
  import tensorflow
  from tensorflow_datasets import testing
  from tensorflow_datasets.core.utils import image_utils
label_desc: Use image node_id 6 for calling the ImageUtilsTest obj's underlying member method code with example usage: obj.test_apply_colormap() without return types