code: def test_dirichlet(self):
      with self.test_session():
          self._test(Dirichlet(tf.zeros(3)), [], [], [3])
          self._test(Dirichlet(tf.zeros([2, 3])), [], [2], [3])
          self._test(
              Dirichlet(tf.zeros(3), sample_shape=1), [1], [], [3]
          )
          self._test(
              Dirichlet(tf.zeros(3), sample_shape=[2, 1]),
              [2, 1],
              [],
              [3],
          )
  
file_path: edward/tests/models/random_variable_shape_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from edward.models import Bernoulli, Dirichlet
label_desc: Use image node_id 3 for calling the test_random_variable_shape_class obj's underlying member method code with example usage: obj.test_dirichlet() without return types