Skip to content

Commit 38ddf93

Browse files
committed
REF: Clean up dMRI data testing file
Clean up dMRI data testing file: - Remove the `_create_dwi_random_dataobj` function as it is no longer necessary. Left behind in commit 63e6f37. - Rename the `_create_dwi_random_data` function to `_dwi_data_to_nifti` to honor its purpose.
1 parent d16c97c commit 38ddf93

File tree

1 file changed

+2
-28
lines changed

1 file changed

+2
-28
lines changed

test/test_data_dmri.py

Lines changed: 2 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -30,33 +30,7 @@
3030
from nifreeze.data.dmri import DWI, find_shelling_scheme, from_nii
3131

3232

33-
def _create_dwi_random_dataobj(request, bvals, bvecs):
34-
rng = request.node.rng
35-
36-
n_gradients = np.count_nonzero(bvals)
37-
b0s = len(bvals) - n_gradients
38-
volumes = n_gradients + b0s
39-
b0_thres = 50
40-
41-
vol_size = (34, 36, 24)
42-
43-
dwi_dataobj = rng.random((*vol_size, volumes), dtype="float32")
44-
affine = np.eye(4, dtype="float32")
45-
brainmask_dataobj = rng.random(vol_size, dtype="float32")
46-
b0_dataobj = rng.random(vol_size, dtype="float32")
47-
gradients = np.vstack([bvecs, bvals[np.newaxis, :]], dtype="float32")
48-
49-
return (
50-
dwi_dataobj,
51-
affine,
52-
brainmask_dataobj,
53-
b0_dataobj,
54-
gradients,
55-
b0_thres,
56-
)
57-
58-
59-
def _create_dwi_random_data(
33+
def _dwi_data_to_nifti(
6034
dwi_dataobj,
6135
affine,
6236
brainmask_dataobj,
@@ -148,7 +122,7 @@ def test_equality_operator(tmp_path, setup_random_dwi_data):
148122
b0_thres,
149123
) = setup_random_dwi_data
150124

151-
dwi, brainmask, b0 = _create_dwi_random_data(
125+
dwi, brainmask, b0 = _dwi_data_to_nifti(
152126
dwi_dataobj,
153127
affine,
154128
brainmask_dataobj,

0 commit comments

Comments
 (0)