Making RNG deterministic

mjbommar · mjbommar · commit 6e9d254b3860 · 2015-03-05T15:41:51.000-05:00
diff --git a/sklearn/tests/test_isotonic.py b/sklearn/tests/test_isotonic.py
@@ -313,13 +313,17 @@ def test_isotonic_zero_weight_loop():
     # Test from @ogrisel's issue:
     # https://github.com/scikit-learn/scikit-learn/issues/4297
 
+    # Get deterministic RNG with seed
+    rng = np.random.RandomState(42)
+
+    # Create regression and samples
     regression = IsotonicRegression()
     n_samples = 50
     x = np.linspace(-3, 3, n_samples)
-    y = x + np.random.uniform(size=n_samples)
+    y = x + rng.uniform(size=n_samples)
 
     # Get some random weights and zero out
-    w = np.random.uniform(size=n_samples)
+    w = rng.uniform(size=n_samples)
     w[5:8] = 0
     regression.fit(x, y, sample_weight=w)
 
