def METHOD_NAME(self):
    positive_matches = tf.random.uniform(
        [2, 1000], minval=0, maxval=1, dtype=tf.float32
    )
    positive_matches = positive_matches > 0.98
    negative_matches = tf.logical_not(positive_matches)
    num_samples = 256
    positive_fraction = 0.25
    _ = balanced_sample(
        positive_matches, negative_matches, num_samples, positive_fraction
    )