test: update testing cases for pypots.utils.random;

WenjieDu · May 24, 2024 · 4c91df4 · 4c91df4
1 parent a405502
commit 4c91df4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/utils/random.py b/tests/utils/random.py
@@ -10,6 +10,7 @@
 import torch
 
 from pypots.utils.random import set_random_seed, get_random_seed
+from pypots.utils.random import RANDOM_SEED as default_random_seed
 
 
 class TestRandom(unittest.TestCase):
@@ -33,8 +34,9 @@ def test_set_random_seed(self):
 
  current_seed = get_random_seed()
  assert (
- not current_seed == 32
- ), "The random seed has been set to 26, not equal to 32."
+ not current_seed == default_random_seed
+ ), f"The random seed has been set to {current_seed} not equal to the default seed {default_random_seed}."
+
  set_random_seed(32)
  current_seed = get_random_seed()
  assert (