code: def testN(self):
      p = [[0.1, 0.2, 0.7], [0.2, 0.3, 0.5]]
      n = [[3.0], [4]]
      with self.cached_session():
          dist = multinomial.Multinomial(total_count=n, probs=p)
          self.assertEqual((2, 1), dist.total_count.get_shape())
          self.assertAllClose(n, dist.total_count)
  
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 4 for calling the MultinomialTest obj's underlying member method code with example usage: obj.testN() without return types