code: def testPdfXStretchedInBroadcastWhenLowerRank(self):
      alpha = [[1.0, 2], [2.0, 3]]
      x = [0.5, 0.5]
      pdf = dirichlet.Dirichlet(alpha).prob(x)
      self.assertAllClose([1.0, 3.0 / 2], self.evaluate(pdf), rtol=1e-5)
      self.assertAllEqual([2], pdf.shape)
  
imports: import numpy
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import exp
  from tensorflow_probability.python.distributions import dirichlet
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.util import deferred_tensor
label_desc: Use image node_id 13 for calling the DirichletTest obj's underlying member method code with example usage: obj.testPdfXStretchedInBroadcastWhenLowerRank() without return types