Week06 tests

canbula · Nov 6, 2023 · ad0c519 · ad0c519
1 parent 5d823b6
commit ad0c519
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Week06/test_pi.py b/Week06/test_pi.py
@@ -1,5 +1,6 @@
 import os
 import numpy as np
+import random
 
 
 files = [f for f in os.listdir(os.path.dirname(__file__)) if f.startswith("pi")]
@@ -26,6 +27,7 @@ def test_generators():
 def test_approaching():
     for f in files:
         np.random.seed(7)
+        random.seed(7)
         gen = eval(f[:-3]).next_pi()
         pi_estimate = next(gen)
         for _ in range(1000):