test_comparisons.py 文件源码

def test_bic_data_dependence():
    d = norm.rvs(size=1000)
    p = norm.logpdf(d)
    c = ChainConsumer()
    c.add_chain(d, posterior=p, num_free_params=1, num_eff_data_points=1000)
    c.add_chain(d, posterior=p, num_free_params=1, num_eff_data_points=500)
    bics = c.comparison.bic()
    assert len(bics) == 2
    assert bics[1] == 0
    expected = np.log(1000) - np.log(500)
    assert np.isclose(bics[0], expected, atol=1e-3)