Set RNG seed for tests.

cvxpy · Sep 16, 2024 · 4a17a65 · 4a17a65
1 parent 2234b2b
commit 4a17a65
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/test_gen_torch_exp.py b/tests/test_gen_torch_exp.py
@@ -8,6 +8,9 @@
 
 from cvxtorch.torch_expression import TorchExpression
 
+seed = 12345
+np.random.seed(seed)
+torch.manual_seed(seed)
 
 class TestGenTorchExp(unittest.TestCase):
     """ Unit tests for the atoms module. """

diff --git a/tests/test_torch_numerics.py b/tests/test_torch_numerics.py
@@ -10,6 +10,9 @@
 from cvxtorch.utils.exp2tch import EXPR2TORCH
 from tests.settings import ATOL, RTOL
 
+seed = 12345
+np.random.seed(seed)
+torch.manual_seed(seed)
 
 class TestTorchNumeric(unittest.TestCase):
     """ Unit tests for the atoms torch_numeric. """