def testZipfSample_ValidateArgs(self):
    power_v = 3.0
    n = int(100000.0)
    for power_dtype in [tf.float32, tf.float64]:
        power = tf.constant(power_v, dtype=power_dtype)
        for dtype in [tf.int32, tf.int64, tf.float32, tf.float64]:
            zipf = tfd.Zipf(power=power, dtype=dtype, validate_args=True)
            samples = zipf.sample(n, seed=test_util.test_seed())
            self.evaluate(samples)
def testZipfSample_ValidateArgs(self):
    power_v = 3.0
    n = int(100000.0)
    for power_dtype in [tf.float32, tf.float64]:
        power = tf.constant(power_v, dtype=power_dtype)
        for dtype in [tf.int32, tf.int64, tf.float32, tf.float64]:
            zipf = tfd.Zipf(power=power, dtype=dtype, validate_args=True)
            samples = zipf.sample(n, seed=test_util.test_seed())
            self.evaluate(samples)