code: def testPmfCountsStretchedInBroadcastWhenLowerRank(self):
      # The probabilities of one vote falling into class k is the mean for class
      # k.
      with self.cached_session():
          alpha = [[1.0, 2], [2.0, 3]]
          counts = [1.0, 0]
          pmf = ds.DirichletMultinomial(1.0, alpha).prob(counts)
          self.assertAllClose([1 / 3.0, 2 / 5.0], self.evaluate(pmf))
          self.assertAllEqual([2], pmf.get_shape())
  
imports: import numpy
  from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops.distributions import dirichlet_multinomial
  from tensorflow.python.platform import test
label_desc: Use image node_id 14 for calling the DirichletMultinomialTest obj's underlying member method code with example usage: obj.testPmfCountsStretchedInBroadcastWhenLowerRank() without return types