code: def test_transform_fn(self):
      context_features, example_features = self._ranker.transform(
          features=self._features, training=False
      )
      self.assertAllEqual(["query_length"], sorted(context_features))
      self.assertAllEqual(
          ["unigrams", "utility"], sorted(example_features)
      )
      self.assertAllEqual(
          [2, 2, 10], example_features["unigrams"].get_shape().as_list()
      )
      self.assertAllEqual([[1], [2]], context_features["query_length"])
      self.assertAllEqual(
          [
              [[tf.math.log1p(1.0)], [0.0]],
              [[0.0], [tf.math.log1p(1.0)]],
          ],
          example_features["utility"],
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow.compat.v2
  from tensorflow_ranking.python.keras import network
label_desc: Use image node_id 2 for calling the BaseRankingNetworkTest obj's underlying member method code with example usage: obj.test_transform_fn() without return types