Addressed comments

Author: seanlaw

stumpy/rng.py

@@ -4,7 +4,7 @@
 
 # Note that an initial SEED = 0 is disallowed
 # in order to account for unit testing
-SEED = np.random.randint(1, 4_294_967_295, dtype=np.uint32)
+SEED = np.random.randint(1, 4_294_967_296, dtype=np.uint32)
 RNG = np.random.RandomState(seed=SEED)
 
 
@@ -47,12 +47,12 @@ def fix_seed(seed):
     -------
     None
     """
-    state = RNG.get_state()
+    curr_state = RNG.get_state()
     RNG.seed(seed)
     try:
         yield
     finally:
-        RNG.set_state(state)
+        RNG.set_state(curr_state)
 
 
 @contextmanager

tests/test_rng.py

@@ -20,6 +20,7 @@ def test_fix_seed():
     with rng.fix_seed(0):
         state = rng.RNG.get_state()
         seed = state[1][0]
+        assert seed == 0
         assert seed != init_seed
 
     state = rng.RNG.get_state()