code: def testPmfNandCountsAgree(self):
      alpha = [[1.0, 2, 3]]
      n = [[5.0]]
      dist = dm.DirichletMultinomial(n, alpha, validate_args=True)
      self.evaluate(dist.prob([2.0, 3, 0]))
      self.evaluate(dist.prob([3.0, 0, 2]))
      with self.assertRaisesOpError("must be non-negative"):
          self.evaluate(dist.prob([-1.0, 4, 2]))
      with self.assertRaisesOpError(
          "last-dimension must sum to `self.total_count`"
      ):
          self.evaluate(dist.prob([3.0, 3, 0]))
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import dirichlet_multinomial
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 6 for calling the DirichletMultinomialTest obj's underlying member method code with example usage: obj.testPmfNandCountsAgree() without return types