code: def testLogits(self):
      p = np.array([[0.1, 0.2, 0.7]], dtype=np.float32)
      logits = np.log(p) - 50.0
      with self.cached_session():
          multinom = multinomial.Multinomial(
              total_count=3.0, logits=logits
          )
          self.assertEqual((1, 3), multinom.probs.get_shape())
          self.assertEqual((1, 3), multinom.logits.get_shape())
          self.assertAllClose(p, multinom.probs)
          self.assertAllClose(logits, multinom.logits)
  
imports: import numpy
  from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops.distributions import multinomial
  from tensorflow.python.platform import test
label_desc: Use image node_id 6 for calling the MultinomialTest obj's underlying member method code with example usage: obj.testLogits() without return types