code: def testZeroCountsResultsInPmfEqualToOne(self):
      # There is only one way for zero items to be selected, and this happens with
      # probability 1.
      alpha = [5, 0.5]
      counts = [0.0, 0]
      dist = dm.DirichletMultinomial(0.0, alpha, validate_args=True)
      pmf = dist.prob(counts)
      self.assertAllClose(1.0, self.evaluate(pmf))
      self.assertEqual((), pmf.shape)
  
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 21 for calling the DirichletMultinomialTest obj's underlying member method code with example usage: obj.testZeroCountsResultsInPmfEqualToOne() without return types