def test_coal(self): a = 5 b = 2 t = 2.0 n = 1.0 for i in range(10): x = coal.sample_coal_cond_counts(a, b, t, n)
def test_sample_coal_cond_counts(self): # test coalescent pdf when conditioned on future lineage counts a = 5 for b in xrange(2, a): t = 500 n = 1000 # draw single coal samples using rejection sampling s = [coal.sample_coal_cond_counts(a, b, t, n) for i in xrange(1000)] eq_sample_pdf(s, lambda x: coal.prob_coal_cond_counts( x, a, b, t, n), 40)