code: def test_categorical_case(self, pmf, fns, rands, targets):
      with self.test_session() as sess:
          for rand, target in zip(rands, targets):
              output = multi_problem_v2.categorical_case(pmf, fns, rand)
              self.assertEqual(sess.run(output), target)
  
file_path: tensor2tensor/tensor2tensor/data_generators/multi_problem_v2_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  from tensor2tensor.data_generators import multi_problem_v2
  from tensor2tensor.data_generators import problem
  import tensorflow.compat.v1
label_desc: Use image node_id 6 for calling the MultiProblemV2Test obj's underlying member method code with example usage: obj.test_categorical_case(pmf, fns, rands, targets) without return types