code: def test_output_categorical_numeric(self):
      type_mapping = {
          types.FeaturePath(["fa"]): schema_pb2.INT,
          types.FeaturePath(["fb"]): schema_pb2.FLOAT,
      }
      self.assertTrue(
          top_k_uniques_stats_util.output_categorical_numeric(
              type_mapping,
              types.FeaturePath(["fa"]),
              statistics_pb2.FeatureNameStatistics.INT,
          )
      )
      self.assertTrue(
          top_k_uniques_stats_util.output_categorical_numeric(
              type_mapping,
              types.FeaturePath(["fb"]),
              statistics_pb2.FeatureNameStatistics.FLOAT,
          )
      )
      self.assertFalse(
          top_k_uniques_stats_util.output_categorical_numeric(
              type_mapping,
              types.FeaturePath(["fc"]),
              statistics_pb2.FeatureNameStatistics.INT,
          )
      )
      self.assertFalse(
          top_k_uniques_stats_util.output_categorical_numeric(
              type_mapping,
              types.FeaturePath(["fb"]),
              statistics_pb2.FeatureNameStatistics.INT,
          )
      )
      self.assertFalse(
          top_k_uniques_stats_util.output_categorical_numeric(
              type_mapping,
              types.FeaturePath(["fa"]),
              statistics_pb2.FeatureNameStatistics.FLOAT,
          )
      )
  
file_path: data-validation/tensorflow_data_validation/utils/top_k_uniques_stats_util_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import absltest
  from tensorflow_data_validation import types
  from tensorflow_data_validation.utils import test_util
  from tensorflow_data_validation.utils import top_k_uniques_stats_util
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 6 for calling the TopKUniquesStatsUtilTest obj's underlying member method code with example usage: obj.test_output_categorical_numeric() without return types