Add HyperGeometricRV

aesara/tensor/random/basic.py

@@ -268,6 +268,17 @@ class GeometricRV(RandomVariable):
 geometric = GeometricRV()
 
 
+class HyperGeometricRV(RandomVariable):
+    name = "hypergeometric"
+    ndim_supp = 0
+    ndims_params = [0, 0, 0]
+    dtype = "int64"
+    _print_name = ("HyperGeom", "\\operatorname{HyperGeom}")
+
+
+hypergeometric = HyperGeometricRV()
+
+
 class CauchyRV(RandomVariable):
     name = "cauchy"
     ndim_supp = 0

tests/tensor/random/test_basic.py

@@ -27,6 +27,7 @@ from aesara.tensor.random.basic import (
     gumbel,
     halfcauchy,
     halfnormal,
+    hypergeometric,
     invgamma,
     laplace,
     multinomial,
@@ -450,6 +451,15 @@ def test_geometric_samples():
     rv_numpy_tester(geometric, test_p, size=[2, 3])
 
 
+def test_hypergeometric_samples():
+    test_ngood = np.array(10, dtype="int64")
+    test_nbad = np.array(20, dtype="int64")
+    test_nsample = np.array(5, dtype="int64")
+
+    rv_numpy_tester(hypergeometric, test_ngood, test_nbad, test_nsample)
+    rv_numpy_tester(hypergeometric, test_ngood, test_nbad, test_nsample, size=[2, 3])
+
+
 def test_cauchy_samples():
     rv_numpy_tester(cauchy, test_fn=stats.cauchy.rvs)