code: def test_dirichlet_categorical(self):
      x_data = np.array([0, 0, 0, 0, 1, 1, 1, 2, 2, 3], np.int32)
      N = x_data.shape[0]
      D = x_data.max() + 1
  
      alpha = np.zeros(D).astype(np.float32) + 2.0
  
      theta = rvs.Dirichlet(alpha)
      x = rvs.Categorical(probs=theta, sample_shape=N)
  
      theta_cond = ed.complete_conditional(theta, [theta, x])
  
      with self.test_session() as sess:
          alpha_val = sess.run(theta_cond.concentration, {x: x_data})
  
      self.assertAllClose(
          alpha_val, np.array([6.0, 5.0, 4.0, 3.0], np.float32)
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import edward
  import numpy
  import tensorflow
  from edward import models
label_desc: Use image node_id 14 for calling the test_conjugacy_class obj's underlying member method code with example usage: obj.test_dirichlet_categorical() without return types