Skip to content

Commit 92555f5

Browse files
authored
Add freesurfer_seed to test_prng_seed function
1 parent 490a89d commit 92555f5

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

petprep/tests/test_config.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,18 +104,21 @@ def test_config_spaces():
104104

105105

106106
@pytest.mark.parametrize(
107-
('master_seed', 'ants_seed', 'numpy_seed'), [(1, 17612, 8272), (100, 19094, 60232)]
107+
('master_seed', 'ants_seed', 'numpy_seed', 'freesurfer_seed'),
108+
[(1, 17612, 8272, 33433), (100, 19094, 60232, 59629)],
108109
)
109-
def test_prng_seed(master_seed, ants_seed, numpy_seed):
110+
def test_prng_seed(master_seed, ants_seed, numpy_seed, freesurfer_seed):
110111
"""Ensure seeds are properly tracked"""
111112
seeds = config.seeds
112113
with patch.dict(os.environ, {}):
113114
seeds.load({'_random_seed': master_seed}, init=True)
114115
assert seeds.master == master_seed
115116
assert seeds.ants == ants_seed
116117
assert seeds.numpy == numpy_seed
118+
assert seeds.freesurfer == freesurfer_seed
117119
assert os.getenv('ANTS_RANDOM_SEED') == str(ants_seed)
120+
assert os.getenv('FREESURFER_RANDOM_SEED') == str(freesurfer_seed)
118121

119122
_reset_config()
120-
for seed in ('_random_seed', 'master', 'ants', 'numpy'):
123+
for seed in ('_random_seed', 'master', 'ants', 'numpy', 'freesurfer'):
121124
assert getattr(config.seeds, seed) is None

0 commit comments

Comments
 (0)