Skip to content

Commit 3d619e7

Browse files
committed
move seed inside parameterized test function
1 parent 060b18e commit 3d619e7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

probscale/tests/test_viz.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
import numpy
55
import 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
910
else:
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
])
512512
def 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

Comments
 (0)