use RandomState instead of seeding whole numpy

epfl-lts2 · Mar 14, 2019 · e46656a · e46656a
1 parent a27c7db
commit e46656a
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/pygsp/tests/test_graphs.py b/pygsp/tests/test_graphs.py
@@ -667,9 +667,9 @@ def key(edge): return str(edge.source()) + ":" + str(edge.target())
 
     def test_networkx_signal_export(self):
         graph = graphs.BarabasiAlbert(N=100, seed=42)
-        np.random.seed(42)
-        signal1 = np.random.random(graph.N)
-        signal2 = np.random.random(graph.N)
+        rs = np.random.RandomState(42)
+        signal1 = rs.normal(size=graph.N)
+        signal2 = rs.normal(size=graph.N)
         graph.set_signal(signal1, "signal1")
         graph.set_signal(signal2, "signal2")
         graph_nx = graph.to_networkx()
@@ -679,9 +679,9 @@ def test_networkx_signal_export(self):
 
     def test_graphtool_signal_export(self):
         g = graphs.Logo()
-        np.random.seed(42)
-        s = np.random.random(g.N)
-        s2 = np.random.random(g.N)
+        rs = np.random.RandomState(42)
+        s = rs.normal(size=g.N)
+        s2 = rs.normal(size=g.N)
         g.set_signal(s, "signal1")
         g.set_signal(s2, "signal2")
         g_gt = g.to_graphtool()
@@ -734,8 +734,7 @@ def test_networkx_signal_import(self):
     def test_save_load(self):
 
         graph = graphs.Sensor(seed=42)
-        np.random.seed(42)
-        signal = np.random.random(graph.N)
+        signal = np.random.RandomState(42).uniform(size=graph.N)
         graph.set_signal(signal, "signal")
 
         # save