code: def testGeomLogPmf(self):
      batch_size = 6
      probs = tf.constant([0.2] * batch_size)
      probs_v = 0.2
      x = np.array([2.0, 3.0, 4.0, 5.0, 6.0, 7.0], dtype=np.float32)
      geom = geometric.Geometric(probs=probs, validate_args=True)
      expected_log_prob = stats.geom.logpmf(x, probs_v, loc=-1)
      log_prob = geom.log_prob(x)
      self.assertEqual(
          [
              6,
          ],
          log_prob.shape,
      )
      self.assertAllClose(expected_log_prob, self.evaluate(log_prob))
  
      pmf = geom.prob(x)
      self.assertEqual(
          [
              6,
          ],
          pmf.shape,
      )
      self.assertAllClose(np.exp(expected_log_prob), self.evaluate(pmf))
  
imports: import numpy
  from scipy import stats
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import geometric
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 3 for calling the GeometricTest obj's underlying member method code with example usage: obj.testGeomLogPmf() without return types