Skip to content

Commit fcda2dc

Browse files
committed
Use "base_fn" instead of "fn" in tests.
This just makes it clearer that that string is the base filename that may or may not be updated.
1 parent 939ccfd commit fcda2dc

File tree

1 file changed

+31
-31
lines changed

1 file changed

+31
-31
lines changed

heudiconv/tests/test_convert.py

Lines changed: 31 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -21,123 +21,123 @@ def test_update_complex_name():
2121
filenames with the part field if appropriate.
2222
"""
2323
# Standard name update
24-
fn = 'sub-X_ses-Y_task-Z_run-01_sbref'
24+
base_fn = 'sub-X_ses-Y_task-Z_run-01_sbref'
2525
metadata = {'ImageType': ['ORIGINAL', 'PRIMARY', 'P', 'MB', 'TE3', 'ND', 'MOSAIC']}
2626
file_counter = 3 # This is the third file with the same name
2727
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_part-phase_sbref'
28-
out_fn_test = update_complex_name(metadata, fn, file_counter)
28+
out_fn_test = update_complex_name(metadata, base_fn, file_counter)
2929
assert out_fn_test == out_fn_true
3030

3131
# Catch an unsupported type and *do not* update
32-
fn = 'sub-X_ses-Y_task-Z_run-01_phase'
33-
out_fn_test = update_complex_name(metadata, fn, file_counter)
34-
assert out_fn_test == fn
32+
base_fn = 'sub-X_ses-Y_task-Z_run-01_phase'
33+
out_fn_test = update_complex_name(metadata, base_fn, file_counter)
34+
assert out_fn_test == base_fn
3535

3636
# Data type is missing from metadata so use suffix
37-
fn = 'sub-X_ses-Y_task-Z_run-01_sbref'
37+
base_fn = 'sub-X_ses-Y_task-Z_run-01_sbref'
3838
metadata = {'ImageType': ['ORIGINAL', 'PRIMARY', 'MB', 'TE3', 'ND', 'MOSAIC']}
3939
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_part-3_sbref'
40-
out_fn_test = update_complex_name(metadata, fn, file_counter)
40+
out_fn_test = update_complex_name(metadata, base_fn, file_counter)
4141
assert out_fn_test == out_fn_true
4242

4343
# Catch existing field with value (part is already in the filename)
4444
# that *does not match* metadata and raise Exception
45-
fn = 'sub-X_ses-Y_task-Z_run-01_part-mag_sbref'
45+
base_fn = 'sub-X_ses-Y_task-Z_run-01_part-mag_sbref'
4646
metadata = {'ImageType': ['ORIGINAL', 'PRIMARY', 'P', 'MB', 'TE3', 'ND', 'MOSAIC']}
4747
file_counter = 3
4848
with pytest.raises(BIDSError):
49-
update_complex_name(metadata, fn, file_counter)
49+
update_complex_name(metadata, base_fn, file_counter)
5050

5151
# Catch existing field with value (part is already in the filename)
5252
# that *does match* metadata and do not update
53-
fn = 'sub-X_ses-Y_task-Z_run-01_part-phase_sbref'
53+
base_fn = 'sub-X_ses-Y_task-Z_run-01_part-phase_sbref'
5454
metadata = {'ImageType': ['ORIGINAL', 'PRIMARY', 'P', 'MB', 'TE3', 'ND', 'MOSAIC']}
5555
file_counter = 3
56-
out_fn_test = update_complex_name(metadata, fn, file_counter)
57-
assert out_fn_test == fn
56+
out_fn_test = update_complex_name(metadata, base_fn, file_counter)
57+
assert out_fn_test == base_fn
5858

5959

6060
def test_update_multiecho_name():
6161
"""Unit testing for heudiconv.convert.update_multiecho_name(), which updates
6262
filenames with the echo field if appropriate.
6363
"""
6464
# Standard name update
65-
fn = 'sub-X_ses-Y_task-Z_run-01_bold'
65+
base_fn = 'sub-X_ses-Y_task-Z_run-01_bold'
6666
metadata = {'EchoTime': 0.01,
6767
'EchoNumber': 1}
6868
echo_times = [0.01, 0.02, 0.03]
6969
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_echo-1_bold'
70-
out_fn_test = update_multiecho_name(metadata, fn, echo_times)
70+
out_fn_test = update_multiecho_name(metadata, base_fn, echo_times)
7171
assert out_fn_test == out_fn_true
7272

7373
# EchoNumber field is missing from metadata, so use echo_times
7474
metadata = {'EchoTime': 0.01}
75-
out_fn_test = update_multiecho_name(metadata, fn, echo_times)
75+
out_fn_test = update_multiecho_name(metadata, base_fn, echo_times)
7676
assert out_fn_test == out_fn_true
7777

7878
# Catch an unsupported type and *do not* update
79-
fn = 'sub-X_ses-Y_task-Z_run-01_phasediff'
80-
out_fn_test = update_multiecho_name(metadata, fn, echo_times)
81-
assert out_fn_test == fn
79+
base_fn = 'sub-X_ses-Y_task-Z_run-01_phasediff'
80+
out_fn_test = update_multiecho_name(metadata, base_fn, echo_times)
81+
assert out_fn_test == base_fn
8282

8383
# EchoTime is missing, but use EchoNumber (which is the first thing it checks)
84-
fn = 'sub-X_ses-Y_task-Z_run-01_bold'
84+
base_fn = 'sub-X_ses-Y_task-Z_run-01_bold'
8585
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_echo-1_bold'
8686
metadata = {'EchoNumber': 1}
8787
echo_times = [False, 0.02, 0.03]
88-
out_fn_test = update_multiecho_name(metadata, fn, echo_times)
88+
out_fn_test = update_multiecho_name(metadata, base_fn, echo_times)
8989
assert out_fn_test == out_fn_true
9090

9191
# Both EchoTime and EchoNumber are missing, which raises a KeyError
92-
fn = 'sub-X_ses-Y_task-Z_run-01_bold'
92+
base_fn = 'sub-X_ses-Y_task-Z_run-01_bold'
9393
metadata = {}
9494
echo_times = [False, 0.02, 0.03]
9595
with pytest.raises(KeyError):
96-
update_multiecho_name(metadata, fn, echo_times)
96+
update_multiecho_name(metadata, base_fn, echo_times)
9797

9898
# Providing echo times as something other than a list should raise a TypeError
99-
fn = 'sub-X_ses-Y_task-Z_run-01_bold'
99+
base_fn = 'sub-X_ses-Y_task-Z_run-01_bold'
100100
with pytest.raises(TypeError):
101-
update_multiecho_name(metadata, fn, set(echo_times))
101+
update_multiecho_name(metadata, base_fn, set(echo_times))
102102

103103

104104
def test_update_uncombined_name():
105105
"""Unit testing for heudiconv.convert.update_uncombined_name(), which updates
106106
filenames with the ch field if appropriate.
107107
"""
108108
# Standard name update
109-
fn = 'sub-X_ses-Y_task-Z_run-01_bold'
109+
base_fn = 'sub-X_ses-Y_task-Z_run-01_bold'
110110
metadata = {'CoilString': 'H1'}
111111
channel_names = ['H1', 'H2', 'H3', 'HEA;HEP']
112112
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_ch-01_bold'
113-
out_fn_test = update_uncombined_name(metadata, fn, channel_names)
113+
out_fn_test = update_uncombined_name(metadata, base_fn, channel_names)
114114
assert out_fn_test == out_fn_true
115115

116116
# CoilString field has no number in it, so we index the channel_names list
117117
metadata = {'CoilString': 'HEA;HEP'}
118118
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_ch-04_bold'
119-
out_fn_test = update_uncombined_name(metadata, fn, channel_names)
119+
out_fn_test = update_uncombined_name(metadata, base_fn, channel_names)
120120
assert out_fn_test == out_fn_true
121121

122122
# Extract the number from the CoilString and use that
123123
channel_names = ['H1', 'B1', 'H3', 'HEA;HEP']
124124
metadata = {'CoilString': 'H1'}
125125
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_ch-01_bold'
126-
out_fn_test = update_uncombined_name(metadata, fn, channel_names)
126+
out_fn_test = update_uncombined_name(metadata, base_fn, channel_names)
127127
assert out_fn_test == out_fn_true
128128

129129
# NOTE: Extracting the number does not protect against multiple coils with the same number
130130
# (but, say, different letters)
131131
# Note that this is still "ch-01"
132132
metadata = {'CoilString': 'B1'}
133133
out_fn_true = 'sub-X_ses-Y_task-Z_run-01_ch-01_bold'
134-
out_fn_test = update_uncombined_name(metadata, fn, channel_names)
134+
out_fn_test = update_uncombined_name(metadata, base_fn, channel_names)
135135
assert out_fn_test == out_fn_true
136136

137137
# Providing echo times as something other than a list should raise a TypeError
138-
fn = 'sub-X_ses-Y_task-Z_run-01_bold'
138+
base_fn = 'sub-X_ses-Y_task-Z_run-01_bold'
139139
with pytest.raises(TypeError):
140-
update_uncombined_name(metadata, fn, set(channel_names))
140+
update_uncombined_name(metadata, base_fn, set(channel_names))
141141

142142

143143
def test_b0dwi_for_fmap(tmpdir, caplog):

0 commit comments

Comments
 (0)