def test_dd_data_prob():
ss = dd.create_ss()
hp = dd.create_hp()
assert_almost_equal(dd.data_prob(hp, ss), 0)
dd.add_data(ss, 0)
assert_almost_equal(dd.data_prob(hp, ss), log(0.5))
def test_dd_add_remove_data():
ss = dd.create_ss()
dd.add_data(ss, 0)
assert ss.counts[0] == 1
assert sum(ss.counts) == 1
dd.remove_data(ss, 0)
assert ss.counts[0] == 0
assert sum(ss.counts) == 0
def test_dd_sample_post():
ss = dd.create_ss()
hp = dd.create_hp()
dd.sample_post(hp, ss)
def test_dd_sample_data():
ss = dd.create_ss()
hp = dd.create_hp()
dd.sample_data(hp, ss)
def test_dd_pred_prob():
ss = dd.create_ss()
hp = dd.create_hp()
assert_almost_equal(dd.pred_prob(hp, ss, 0), log(0.5))
assert_almost_equal(dd.pred_prob(hp, ss, 1), log(0.5))
def test_dd_create_ss():
ss = dd.create_ss()
assert len(ss.counts) == 2
