Make unit tests use a common seed function

pybamm-team · Oct 3, 2023 · 853f920 · 853f920
1 parent 1b4d444
commit 853f920
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/testcase.py b/tests/testcase.py
@@ -3,10 +3,11 @@
 #
 import unittest
 import hashlib
-import numpy as np
 from functools import wraps
 from types import FunctionType
 
+import pybamm
+
 
 def FixRandomSeed(method):
     """
@@ -23,7 +24,7 @@ def FixRandomSeed(method):
 
     @wraps(method)
     def wrapped(*args, **kwargs):
-        np.random.seed(
+        pybamm.util.set_random_seed(
             int(hashlib.sha256(method.__name__.encode()).hexdigest(), 16) % (2**32)
         )
         return method(*args, **kwargs)