Add VonMisesRV

aesara/tensor/random/basic.py

@@ -204,6 +204,17 @@ class LogisticRV(RandomVariable):
 logistic = LogisticRV()
 
 
+class VonMisesRV(RandomVariable):
+    name = "vonmises"
+    ndim_supp = 0
+    ndims_params = [0, 0]
+    dtype = "floatX"
+    _print_name = ("VonMises", "\\operatorname{VonMises}")
+
+
+vonmises = VonMisesRV()
+
+
 def safe_multivariate_normal(mean, cov, size=None, rng=None):
     """A shape consistent multivariate normal sampler.
 

tests/tensor/random/test_basic.py

@@ -44,6 +44,7 @@ from aesara.tensor.random.basic import (
     triangular,
     truncexpon,
     uniform,
+    vonmises,
     wald,
     weibull,
 )
@@ -277,6 +278,14 @@ def test_logistic_samples():
     rv_numpy_tester(logistic, test_loc, test_scale, size=[2, 3])
 
 
+def test_vonmises_samples():
+    test_mu = np.array(np.pi, dtype=config.floatX)
+    test_kappa = np.array(0.5, dtype=config.floatX)
+
+    rv_numpy_tester(vonmises, test_mu, test_kappa)
+    rv_numpy_tester(vonmises, test_mu, test_kappa, size=[2, 3])
+
+
 def test_pareto_samples():
     test_alpha = np.array(0.5, dtype=config.floatX)