|
10 | 10 |
|
11 | 11 | @pytest.fixture(scope="module")
|
12 | 12 | def cifti_data():
|
13 |
| - out = Path().absolute() |
14 |
| - volume_file = str(out / "volume.nii.gz") |
15 |
| - left_gii = str(out / "left.gii") |
16 |
| - right_gii = str(out / "right.gii") |
17 |
| - surface_data = [nb.gifti.GiftiDataArray(np.ones(32492)) for _ in range(4)] |
18 |
| - vol = nb.Nifti1Image(np.ones((91, 109, 91, 4)), np.eye(4)) |
19 |
| - gii = nb.GiftiImage(darrays=surface_data) |
20 |
| - |
21 |
| - vol.to_filename(volume_file) |
22 |
| - gii.to_filename(left_gii) |
23 |
| - gii.to_filename(right_gii) |
24 |
| - yield volume_file, left_gii, right_gii |
25 |
| - for f in (volume_file, left_gii, right_gii): |
26 |
| - Path(f).unlink() |
| 13 | + import tempfile |
| 14 | + |
| 15 | + with tempfile.TemporaryDirectory('cifti-data') as tmp: |
| 16 | + out = Path(tmp).absolute() |
| 17 | + volume_file = str(out / "volume.nii.gz") |
| 18 | + left_gii = str(out / "left.gii") |
| 19 | + right_gii = str(out / "right.gii") |
| 20 | + surface_data = [nb.gifti.GiftiDataArray(np.ones(32492)) for _ in range(4)] |
| 21 | + vol = nb.Nifti1Image(np.ones((91, 109, 91, 4)), np.eye(4)) |
| 22 | + gii = nb.GiftiImage(darrays=surface_data) |
| 23 | + |
| 24 | + vol.to_filename(volume_file) |
| 25 | + gii.to_filename(left_gii) |
| 26 | + gii.to_filename(right_gii) |
| 27 | + yield volume_file, left_gii, right_gii |
27 | 28 |
|
28 | 29 |
|
29 | 30 | def test_GenerateCifti(tmpdir, cifti_data):
|
|
0 commit comments