code: def testPdfXStretchedInBroadcastWhenSameRank(self):
      a = [[1.0, 2], [2.0, 3]]
      b = [[1.0, 2], [2.0, 3]]
      x = [[0.5, 0.5]]
      pdf = kumaraswamy.Kumaraswamy(a, b, validate_args=True).prob(x)
      expected_pdf = _kumaraswamy_pdf(a, b, x)
      self.assertAllClose(expected_pdf, self.evaluate(pdf))
      self.assertEqual((2, 2), pdf.shape)
  
imports: import numpy
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kumaraswamy
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 12 for calling the KumaraswamyTest obj's underlying member method code with example usage: obj.testPdfXStretchedInBroadcastWhenSameRank() without return types