code: def testUniformSampleWithShape(self):
      a = 10.0
      b = [11.0, 20.0]
      dist = uniform.Uniform(a, b, validate_args=True)
  
      pdf = dist.prob(dist.sample((2, 3), seed=test_util.test_seed()))
      # pylint: disable=bad-continuation
      expected_pdf = [
          [[1.0, 0.1], [1.0, 0.1], [1.0, 0.1]],
          [[1.0, 0.1], [1.0, 0.1], [1.0, 0.1]],
      ]
      # pylint: enable=bad-continuation
      self.assertAllClose(expected_pdf, self.evaluate(pdf))
  
      pdf = dist.prob(dist.sample(seed=test_util.test_seed()))
      expected_pdf = [1.0, 0.1]
      self.assertAllClose(expected_pdf, self.evaluate(pdf))
  
imports: import numpy
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import uniform
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 17 for calling the UniformTest obj's underlying member method code with example usage: obj.testUniformSampleWithShape() without return types