class_docstr: Tests for TopKUniquesSketchStatsGenerator.
class_docstr_tok: ["Tests","for","TopKUniquesSketchStatsGenerator","."]
code: def test_topk_uniques_sketch_with_int_weights(self):
      # non-weighted ordering
      # 3 'a', 2 'e', 2 'd', 2 'c', 1 'b'
      # weighted ordering
      # fa: 20 'e', 20 'd', 15 'a', 10 'c', 5 'b'
      batches = [
          pa.RecordBatch.from_arrays(
              [
                  pa.array(
                      [["a", "b", "c", "e"], ["a", "c", "d", "a"]],
                      type=pa.list_(pa.binary()),
                  ),
                  pa.array([[5], [5]], type=pa.list_(pa.int32())),
              ],
              ["fa", "w"],
          ),
          pa.RecordBatch.from_arrays(
              [
                  pa.array([["d", "e"]], type=pa.list_(pa.binary())),
                  pa.array([[15]], type=pa.list_(pa.int32())),
              ],
              ["fa", "w"],
          ),
      ]
      expected_result = {
          types.FeaturePath(["fa"]): text_format.Parse(
              """
                  path {
                    step: 'fa'
                  }
                  string_stats {
                    unique: 5
                    top_values {
                      value: 'a'
                      frequency: 3.0
                    }
                    top_values {
                      value: 'e'
                      frequency: 2.0
                    }
                    top_values {
                      value: 'd'
                      frequency: 2.0
                    }
                    top_values {
                      value: 'c'
                      frequency: 2.0
                    }
                    rank_histogram {
                      buckets {
                        low_rank: 0
                        high_rank: 0
                        label: "a"
                        sample_count: 3.0
                      }
                      buckets {
                        low_rank: 1
                        high_rank: 1
                        label: "e"
                        sample_count: 2.0
                      }
                      buckets {
                        low_rank: 2
                        high_rank: 2
                        label: "d"
                        sample_count: 2.0
                      }
                    }
                    weighted_string_stats {
                      top_values {
                        value: 'e'
                        frequency: 20.0
                      }
                      top_values {
                        value: 'd'
                        frequency: 20.0
                      }
                      top_values {
                        value: 'a'
                        frequency: 15.0
                      }
                      top_values {
                        value: 'c'
                        frequency: 10.0
                      }
                      rank_histogram {
                        buckets {
                          low_rank: 0
                          high_rank: 0
                          label: "e"
                          sample_count: 20.0
                        }
                        buckets {
                          low_rank: 1
                          high_rank: 1
                          label: "d"
                          sample_count: 20.0
                        }
                        buckets {
                          low_rank: 2
                          high_rank: 2
                          label: "a"
                          sample_count: 15.0
                        }
                      }
                    }
                }""",
              statistics_pb2.FeatureNameStatistics(),
          )
      }
      generator = sketch_generator.TopKUniquesSketchStatsGenerator(
          example_weight_map=ExampleWeightMap(weight_feature="w"),
          num_top_values=4,
          num_rank_histogram_buckets=3,
      )
      self.assertCombinerOutputEqual(
          batches, generator, expected_result
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import absltest
  from absl.testing import parameterized
  import pyarrow
  from tensorflow_data_validation import types
  from tensorflow_data_validation.statistics.generators import top_k_uniques_sketch_stats_generator
  from tensorflow_data_validation.utils import test_util
  from tensorflow_data_validation.utils.example_weight_map import ExampleWeightMap
  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 12 for calling the TopKUniquesSketchStatsGeneratorTest obj's underlying member method code with example usage: obj.test_topk_uniques_sketch_with_int_weights() without return types