def binomial_log_pdf_ddd(k, p, n):
from qy.math import ln_choose
@qy.if_(k > n)
def _():
qy.return_(-numpy.inf)
@qy.if_(p == 0.0)
def _():
qy.return_(qy.select(k == 0.0, 0.0, -numpy.inf))
@qy.if_(p == 1.0)
def _():
qy.return_(qy.select(k == n, 0.0, -numpy.inf))
qy.return_(ln_choose(n, k) + k * qy.log(p) + (n - k) * qy.log1p(-p))
def binomial_log_pdf_ddd(k, p, n):
from qy.math import ln_choose
@qy.if_(k > n)
def _():
qy.return_(-numpy.inf)
@qy.if_(p == 0.0)
def _():
qy.return_(qy.select(k == 0.0, 0.0, -numpy.inf))
@qy.if_(p == 1.0)
def _():
qy.return_(qy.select(k == n, 0.0, -numpy.inf))
qy.return_(
ln_choose(n, k) + k * qy.log(p) + (n - k) * qy.log1p(-p))
def assert_ln_choose_ok(n, k):
@qy.python(ln_choose(n, k))
def _(v):
assert_almost_equal(v, log(choose(n, k)))
