|
54 | 54 | Triangular,
|
55 | 55 | TruncatedNormal,
|
56 | 56 | Uniform,
|
| 57 | + VonMises, |
57 | 58 | Wald,
|
58 | 59 | Weibull,
|
59 | 60 | ZeroInflatedBinomial,
|
@@ -437,6 +438,20 @@ def test_pareto_moment(alpha, m, size, expected):
|
437 | 438 | Pareto("x", alpha=alpha, m=m, size=size)
|
438 | 439 | assert_moment_is_expected(model, expected)
|
439 | 440 |
|
| 441 | +@pytest.mark.parametrize( |
| 442 | + "mu, kappa, size, expected", |
| 443 | + [ |
| 444 | + (0, 1, None, 0), |
| 445 | + (0, np.ones(4), None, np.zeros(4)), |
| 446 | + (np.arange(4), 0.5, None, np.arange(4)), |
| 447 | + (np.arange(4), np.arange(1, 5), (2, 4), np.full((2, 4), np.arange(4))), |
| 448 | + ], |
| 449 | +) |
| 450 | +def test_vonmises_moment(mu, kappa, size, expected): |
| 451 | + with Model() as model: |
| 452 | + VonMises("x", mu=mu, kappa=kappa, size=size) |
| 453 | + assert_moment_is_expected(model, expected) |
| 454 | + |
440 | 455 |
|
441 | 456 | @pytest.mark.parametrize(
|
442 | 457 | "mu, lam, phi, size, expected",
|
|
0 commit comments