Add GeometricRV

aesara/tensor/random/basic.py

@@ -257,6 +257,17 @@ class PoissonRV(RandomVariable):
 poisson = PoissonRV()
 
 
+class GeometricRV(RandomVariable):
+    name = "geometric"
+    ndim_supp = 0
+    ndims_params = [0]
+    dtype = "int64"
+    _print_name = ("Geom", "\\operatorname{Geom}")
+
+
+geometric = GeometricRV()
+
+
 class CauchyRV(RandomVariable):
     name = "cauchy"
     ndim_supp = 0

tests/tensor/random/test_basic.py

@@ -23,6 +23,7 @@ from aesara.tensor.random.basic import (
     dirichlet,
     exponential,
     gamma,
+    geometric,
     gumbel,
     halfcauchy,
     halfnormal,
@@ -442,6 +443,13 @@ def test_poisson_samples():
     rv_numpy_tester(poisson, test_lambda, size=[2, 3])
 
 
+def test_geometric_samples():
+    test_p = np.array(0.1, dtype=config.floatX)
+
+    rv_numpy_tester(geometric, test_p)
+    rv_numpy_tester(geometric, test_p, size=[2, 3])
+
+
 def test_cauchy_samples():
     rv_numpy_tester(cauchy, test_fn=stats.cauchy.rvs)