code: def test_prob(self):
      batch_ps = []
      for a in range(self.n):
          indices = np.asarray([a] * self.batch_size, dtype=np.int32)
          batch_p = self.distrib.prob(indices)
          self.assertTrue(isinstance(batch_p, chainer.Variable))
          for b in range(self.batch_size):
              p = batch_p.array[b]
              self.assertGreaterEqual(p, self.min_prob)
              self.assertLessEqual(p, 1)
          batch_ps.append(batch_p.array)
      np.testing.assert_almost_equal(
          sum(batch_ps), np.ones(self.batch_size)
      )
  
imports: import unittest
  import chainer
  from chainer import testing
  from chainer.testing import condition
  import numpy
  import scipy.stats
  from chainerrl import distribution
label_desc: Use image node_id 3 for calling the TestSoftmaxDistribution obj's underlying member method code with example usage: obj.test_prob() without return types