code: def test_docstring_example(self):
      stream = test_util.test_seed_stream()
      loc = random_ops.spherical_uniform([10], 3, seed=stream())
      components_dist = von_mises_fisher.VonMisesFisher(
          mean_direction=loc, concentration=50.0
      )
      mixture_dist = categorical.Categorical(
          logits=tf.random.uniform([500, 10], seed=stream())
      )
      obs_dist = mixture_same_family.MixtureSameFamily(
          mixture_dist,
          batch_broadcast.BatchBroadcast(components_dist, [500, 10]),
      )
      test_sites = random_ops.spherical_uniform([20], 3, seed=stream())
      lp = sample_lib.Sample(obs_dist, 20).log_prob(test_sites)
      self.assertEqual([500], lp.shape)
      self.evaluate(lp)
  
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 12 for calling the _BatchBroadcastTest obj's underlying member method code with example usage: obj.test_docstring_example() without return types