def METHOD_NAME(self):
    positive_matches = tf.constant(
        [
            False,
            False,
            False,
            False,
            False,
            False,
            False,
            False,
            False,
            False,
        ]
    )
    # the rest are neither positive nor negative, but ignored matches
    negative_matches = tf.constant(
        [False, False, True, False, False, True, False, False, True, False]
    )
    num_samples = 5
    positive_fraction = 0.5
    res = balanced_sample(
        positive_matches, negative_matches, num_samples, positive_fraction
    )
    # given only 3 negative and 0 positive, select all of them
    self.assertAllClose(res, [0, 0, 1, 0, 0, 1, 0, 0, 1, 0])