class_docstr: Tests for TopKUniquesSketchStatsGenerator.
class_docstr_tok: ["Tests","for","TopKUniquesSketchStatsGenerator","."]
code: def test_topk_with_frequency_threshold(self):
      batches = [
          pa.RecordBatch.from_arrays(
              [
                  pa.array([["a", "b", "y", "b"]]),
                  pa.array([[5.0]]),
              ],
              ["fa", "w"],
          ),
          pa.RecordBatch.from_arrays(
              [
                  pa.array([["a", "x", "a", "z"]]),
                  pa.array([[15.0]]),
              ],
              ["fa", "w"],
          ),
      ]
      expected_result = {
          types.FeaturePath(["fa"]): text_format.Parse(
              """
          path {
            step: 'fa'
          }
          string_stats {
            unique: 5
            top_values {
              value: 'a'
              frequency: 3
            }
            top_values {
              value: 'b'
              frequency: 2
            }
            rank_histogram {
              buckets {
                low_rank: 0
                high_rank: 0
                label: "a"
                sample_count: 3.0
              }
              buckets {
                low_rank: 1
                high_rank: 1
                label: "b"
                sample_count: 2.0
              }
            }
            weighted_string_stats {
              top_values {
                value: 'a'
                frequency: 35.0
              }
              top_values {
                value: 'z'
                frequency: 15.0
              }
              top_values {
                value: 'x'
                frequency: 15.0
              }
              rank_histogram {
                buckets {
                  low_rank: 0
                  high_rank: 0
                  label: "a"
                  sample_count: 35.0
                }
                buckets {
                  low_rank: 1
                  high_rank: 1
                  label: "z"
                  sample_count: 15.0
                }
                buckets {
                  low_rank: 2
                  high_rank: 2
                  label: "x"
                  sample_count: 15.0
                }
              }
            }
          }""",
              statistics_pb2.FeatureNameStatistics(),
          )
      }
  
      generator = sketch_generator.TopKUniquesSketchStatsGenerator(
          example_weight_map=ExampleWeightMap(weight_feature="w"),
          num_top_values=5,
          frequency_threshold=2,
          weighted_frequency_threshold=15,
          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 10 for calling the TopKUniquesSketchStatsGeneratorTest obj's underlying member method code with example usage: obj.test_topk_with_frequency_threshold() without return types