code: def testPmfandCountsAgree(self):
      p = [[0.1, 0.2, 0.7]]
      n = [[5.0]]
      with self.cached_session():
          dist = multinomial.Multinomial(
              total_count=n, probs=p, validate_args=True
          )
          dist.prob([2.0, 3, 0]).eval()
          dist.prob([3.0, 0, 2]).eval()
          with self.assertRaisesOpError("must be non-negative"):
              dist.prob([-1.0, 4, 2]).eval()
          with self.assertRaisesOpError(
              "counts must sum to `self.total_count`"
          ):
              dist.prob([3.0, 3, 0]).eval()
  
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 8 for calling the MultinomialTest obj's underlying member method code with example usage: obj.testPmfandCountsAgree() without return types