44import numpy
55import matplotlib .pyplot as plt
66
7- if sys .version_info .major == 2 : # pragma: no cover
7+ PY2K = sys .version_info .major == 2
8+ if PY2K : # pragma: no cover
89 import mock
910else :
1011 from unittest import mock
@@ -498,7 +499,6 @@ def test__fit_simple(plot_data, fitlogs, known_yhat):
498499 nptest .assert_allclose (yhat , known_yhat , rtol = 0.0001 )
499500
500501
501- @seed
502502@pytest .mark .parametrize (('fitlogs' , 'known_lo' , 'known_hi' ), [
503503 (None , numpy .array ([- 0.7944 , 2.7051 , 6.1974 , 9.2612 , 11.9382 , 14.4290 ]),
504504 numpy .array ([ 2.1447 , 4.8360 , 7.7140 , 10.8646 , 14.1014 , 17.4432 ])),
@@ -510,6 +510,7 @@ def test__fit_simple(plot_data, fitlogs, known_yhat):
510510 numpy .array ([5.5107 , 13.0148 , 17.232 , 20.4285 , 23.1035 , 25.3843 ])),
511511])
512512def test__fit_ci (plot_data , fitlogs , known_lo , known_hi ):
513+ numpy .random .seed (0 )
513514 x = numpy .arange (1 , len (plot_data )+ 1 )
514515 xhat = x [::6 ]
515516 yhat_lo , yhat_hi = viz ._fit_ci (x , plot_data , xhat , fitlogs = fitlogs , niter = 1000 )
0 commit comments