code: def _test(self, n, *args, **kwargs):
      rv = ParamMixture(*args, **kwargs)
      val_est = rv.sample(n).shape
      val_true = (
          tf.TensorShape(n)
          .concatenate(rv.cat.batch_shape)
          .concatenate(rv.components.event_shape)
      )
      self.assertEqual(val_est, val_true)
  
      self.assertEqual(rv.sample_shape, rv.cat.sample_shape)
      self.assertEqual(rv.sample_shape, rv.components.sample_shape)
      self.assertEqual(rv.batch_shape, rv.cat.batch_shape)
      self.assertEqual(rv.event_shape, rv.components.event_shape)
  
file_path: edward/tests/models/param_mixture_sample_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import edward
  import numpy
  import tensorflow
  from edward.models import Beta, Dirichlet, Normal, ParamMixture
label_desc: Use image node_id 1 for calling the test_param_mixture_sample_class obj's underlying member method code with example usage: obj._test(n) without return types