Skip to content

Commit 547dafd

Browse files
author
Michael Dayan
committed
Set label seed for the whole test suite and print it to stdout
1 parent 457bf40 commit 547dafd

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

heudiconv/tests/test_bids.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,11 +50,13 @@
5050

5151
import pytest
5252

53-
def gen_rand_label(label_size, label_seed):
53+
def gen_rand_label(label_size, label_seed, seed_stdout=True):
5454
seed(label_seed)
5555
rand_char = ''.join(choice(string.ascii_letters) for _ in range(label_size-1))
5656
seed(label_seed)
5757
rand_num = choice(string.digits)
58+
if seed_stdout:
59+
print(f'Seed used to generate custom label: {label_seed}')
5860
return rand_char + rand_num
5961

6062
def test_maybe_na():
@@ -78,7 +80,7 @@ def test_treat_age():
7880

7981
SHIM_LENGTH = 6
8082
TODAY = datetime.today()
81-
83+
LABEL_SEED = int.from_bytes(os.urandom(8), byteorder="big")
8284

8385
A_SHIM = [random() for i in range(SHIM_LENGTH)]
8486
def test_get_shim_setting(tmpdir):
@@ -97,7 +99,7 @@ def test_get_shim_setting(tmpdir):
9799
assert get_shim_setting(json_name) == A_SHIM
98100

99101

100-
def test_get_key_info_for_fmap_assignment(tmpdir, label_size=4, label_seed=42):
102+
def test_get_key_info_for_fmap_assignment(tmpdir, label_size=4, label_seed=LABEL_SEED):
101103
"""
102104
Test get_key_info_for_fmap_assignment.
103105
@@ -532,7 +534,7 @@ def create_dummy_no_shim_settings_bids_session(session_path):
532534

533535
return session_struct, expected_result, expected_fmap_groups, expected_compatible_fmaps
534536

535-
def create_dummy_no_shim_settings_custom_label_bids_session(session_path, label_size=4, label_seed=42):
537+
def create_dummy_no_shim_settings_custom_label_bids_session(session_path, label_size=4, label_seed=LABEL_SEED):
536538
"""
537539
Creates a dummy BIDS session, with slim json files and empty nii.gz
538540
The fmap files are pepolar

0 commit comments

Comments
 (0)