diff --git a/test_convoys.py b/test_convoys.py index a53a441..ad7e0dd 100644 --- a/test_convoys.py +++ b/test_convoys.py @@ -50,12 +50,12 @@ def test_exponential_regression_model(c=0.3, lambd=0.1, n=100000): assert 0.70*(c_hi-c_lo) < (y_hi-y_lo) < 1.30*(c_hi-c_lo) # Check the random variates - will_convert, convert_at = model.rvs([1], n_curves=1, n_samples=1000) + will_convert, convert_at = model.rvs([1], n_curves=1, n_samples=10000) assert 0.95*c < numpy.mean(will_convert) < 1.05*c convert_times = convert_at[will_convert] for t in [1, 3, 10]: d = 1 - numpy.exp(-lambd*t) - assert 0.9*d < (convert_times < t).mean() < 1.1*d + assert 0.8*d < (convert_times < t).mean() < 1.2*d @flaky.flaky