code: def testInternalShapeInference(self):
      # Regression test for the effect of b/139013403 on vMF sampling.
      sample_shape = tf.constant([2])
      # There needs to be a 1 dimension in the batch shape to trigger the bug
      mean_dir = tf.math.l2_normalize([1.0, 2, 3, 4], axis=-1)
      concentration = [0]
      vmf = von_mises_fisher.VonMisesFisher(
          mean_direction=mean_dir,
          concentration=concentration,
          validate_args=True,
          allow_nan_stats=False,
      )
      self.evaluate(
          vmf.sample(sample_shape, seed=test_util.test_seed())
      )
  
imports: from absl.testing import parameterized
  import numpy
  from scipy import special
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import spherical_uniform
  from tensorflow_probability.python.distributions import von_mises_fisher
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 13 for calling the VonMisesFisherTest obj's underlying member method code with example usage: obj.testInternalShapeInference() without return types