code: def test_docstring_shapes(self):
      d = batch_broadcast.BatchBroadcast(
          normal.Normal(tf.range(3.0), 1.0), [2, 3]
      )
      self.assertEqual([2, 3], d.batch_shape)
      self.assertEqual([3], d.distribution.batch_shape)
      self.assertEqual([], d.event_shape)
  
      df = uniform.Uniform(4.0, 5.0).sample(
          [10, 1], seed=test_util.test_seed()
      )
      d = batch_broadcast.BatchBroadcast(
          wishart.WishartTriL(df=df, scale_tril=tf.eye(3)), [2]
      )
      self.assertEqual([10, 2], d.batch_shape)
      self.assertEqual([10, 1], d.distribution.batch_shape)
      self.assertEqual([3, 3], d.event_shape)
  
imports: import hypothesis
  import hypothesis.strategies
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import batch_broadcast
  from tensorflow_probability.python.distributions import bernoulli
  from tensorflow_probability.python.distributions import beta
  from tensorflow_probability.python.distributions import categorical
  from tensorflow_probability.python.distributions import dirichlet
  from tensorflow_probability.python.distributions import hypothesis_testlib
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import mixture_same_family
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import sample
  from tensorflow_probability.python.distributions import uniform
  from tensorflow_probability.python.distributions import von_mises_fisher
  from tensorflow_probability.python.distributions import wishart
  from tensorflow_probability.python.internal import hypothesis_testlib
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.random import random_ops
label_desc: Use image node_id 11 for calling the _BatchBroadcastTest obj's underlying member method code with example usage: obj.test_docstring_shapes() without return types