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
