def test_prob_coal2(self): outdir = 'test/tmp/test_coal/Coal_test_prob_coal2/' make_clean_dir(outdir) k = 2 n = 1000 p = Gnuplot() p.enableOutput(False) p.plotfunc(lambda t: coal.prob_coal(t, k, n), 0, 4000, 10, ymin=0) # draw single coal samples x = [coal.sample_coal(k, n) for i in xrange(200)] plotdistrib(x, 40, plot=p) p.enableOutput(True) p.save(outdir + 'plot.png') eq_sample_pdf(x, lambda t: coal.prob_coal(t, k, n), 40)
def test_plot_bounded_coal(self): n = 1000 k = 6 T = 500 # plots should differ p = plotfunc(lambda t: coal.prob_bounded_coal(t, k, n, T), 0, 1000, 10) p.plotfunc(lambda t: coal.prob_coal(t, k, n), 0, 1000, 10)