class_docstr: Tests for util functions.
code: def test_get_params(self):
      params = {
          "num_shuffles_train": 1,
          "num_shuffles_eval": 2,
          "num_shuffles_predict": 3,
      }
      self.assertEqual(
          model._get_params(tf_estimator.ModeKeys.TRAIN, params), 1
      )
      self.assertEqual(
          model._get_params(tf_estimator.ModeKeys.EVAL, params), 2
      )
      self.assertEqual(
          model._get_params(tf_estimator.ModeKeys.PREDICT, params), 3
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  from absl.testing import parameterized
  import tensorflow
  from tensorflow import estimator
  from tensorflow_ranking.python import feature
  from tensorflow_ranking.python import head
  from tensorflow_ranking.python import losses
  from tensorflow_ranking.python import model
label_desc: Use image node_id 1 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_get_params() without return types