def test_ztnbinom_pmf(self):
"Test ztnbinom.pmf"
(n, p) = (.3, .15)
v = np.arange(1, 21)
vals = ztnbinom(n, p).pmf(v)
ctrl = dist.nbinom(n, p).pmf(v) / (1. - dist.nbinom(n, p).pmf(0))
assert_almost_equal(vals, ctrl)
#
vals = ztnbinom(n, p).cdf(v)
control = [ztnbinom(n, p).pmf(np.arange(1, _ + 1)).sum() for _ in v]
assert_almost_equal(vals, control)
#
assert_equal(ztnbinom(n, p).ppf(vals), v)
#
(n, p) = (1.5, .15)
v = np.arange(1, 21)
vals = ztnbinom(n, p).pmf(v)
ctrl = dist.nbinom(n, p).pmf(v) / (1. - dist.nbinom(n, p).pmf(0))
assert_almost_equal(vals, ctrl)
#
vals = ztnbinom(n, p).cdf(v)
control = [ztnbinom(n, p).pmf(np.arange(1, _ + 1)).sum() for _ in v]
assert_almost_equal(vals, control)
#
assert_equal(ztnbinom(n, p).ppf(vals), v)
def test_ztnbinom_pmf(self):
"Test ztnbinom.pmf"
(n, p) = (.3, .15)
v = np.arange(1, 21)
vals = ztnbinom(n, p).pmf(v)
ctrl = dist.nbinom(n, p).pmf(v)/(1.-dist.nbinom(n, p).pmf(0))
assert_almost_equal(vals, ctrl)
#
vals = ztnbinom(n, p).cdf(v)
control = [ztnbinom(n, p).pmf(np.arange(1, _+1)).sum() for _ in v]
assert_almost_equal(vals, control)
#
assert_equal(ztnbinom(n, p).ppf(vals), v)
#
(n, p) = (1.5, .15)
v = np.arange(1, 21)
vals = ztnbinom(n, p).pmf(v)
ctrl = dist.nbinom(n, p).pmf(v)/(1.-dist.nbinom(n, p).pmf(0))
assert_almost_equal(vals, ctrl)
#
vals = ztnbinom(n, p).cdf(v)
control = [ztnbinom(n, p).pmf(np.arange(1, _+1)).sum() for _ in v]
assert_almost_equal(vals, control)
#
assert_equal(ztnbinom(n, p).ppf(vals), v)
def test_basic(self):
(n, p) = (.3, .15)
v = np.arange(0, 21)
vals = dist.nbinom(n, p).cdf(v)
control = [dist.nbinom(n, p).pmf(np.arange(_ + 1)).sum() for _ in v]
assert_almost_equal(vals, control)
def test_basic(self):
(n, p) = (.3, .15)
v = np.arange(0, 21)
vals = dist.nbinom(n, p).cdf(v)
control = [dist.nbinom(n, p).pmf(np.arange(_+1)).sum() for _ in v]
assert_almost_equal(vals, control)