Ejemplo n.º 1
0
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))
Ejemplo n.º 2
0
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