code: def testAssertValidSample(self):
      logits = [0.1, 0.4, 0.5, 0.3, 0.1]
      dist = oc.OneHotCategorical(logits, validate_args=True)
      with self.assertRaisesOpError("Condition x >= 0"):
          self.evaluate(dist.prob([-1.0, 0.0, 1.0, 1.0, 0.0]))
      with self.assertRaisesOpError(
          "Last dimension of sample must sum to 1."
      ):
          self.evaluate(dist.log_prob([1.0, 0.0, 1.0, 0.0, 0.0]))
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import identity
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import onehot_categorical
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.util import deferred_tensor
label_desc: Use image node_id 9 for calling the OneHotCategoricalTest obj's underlying member method code with example usage: obj.testAssertValidSample() without return types