def test_predict_prob(self):
res = self.res
pr = res.predict(which='prob')
pr2 = truncatedpoisson.pmf(
np.arange(8), res.predict(which="mean-main")[:, None], 0)
assert_allclose(pr, pr2, rtol=1e-10, atol=1e-10)
def test_pmf(self):
poisson_pmf = poisson.pmf(4, 6) / (1 - poisson.cdf(2, 6))
tpoisson_pmf = truncatedpoisson.pmf(4, 6, 2)
assert_allclose(poisson_pmf, tpoisson_pmf, rtol=1e-7)
def test_pmf_zero(self):
poisson_pmf = poisson.pmf(2, 2) / poisson.sf(0, 2)
tpoisson_pmf = truncatedpoisson.pmf(2, 2, 0)
assert_allclose(poisson_pmf, tpoisson_pmf, rtol=1e-7)