|
6 | 6 | from ..imagemanip import copy_header |
7 | 7 |
|
8 | 8 |
|
9 | | -@pytest.mark.parametrize('keep_dtype', (True, False)) |
| 9 | +@pytest.mark.parametrize("keep_dtype", (True, False)) |
10 | 10 | def test_copy_header(tmp_path, keep_dtype): |
11 | 11 | """Cover copy_header.""" |
12 | | - fname1 = tmp_path / 'reference.nii.gz' |
13 | | - fname2 = tmp_path / 'target.nii.gz' |
| 12 | + fname1 = tmp_path / "reference.nii.gz" |
| 13 | + fname2 = tmp_path / "target.nii.gz" |
14 | 14 |
|
15 | | - nii = nb.Nifti1Image( |
16 | | - np.zeros((10, 10, 10), dtype='uint8'), None, None) |
17 | | - nii.set_qform(np.diag((1., 2., 3., 1.)), code=2) |
18 | | - nii.set_sform(np.diag((1., 2., 3., 1.)), code=1) |
| 15 | + nii = nb.Nifti1Image(np.zeros((10, 10, 10), dtype="uint8"), None, None) |
| 16 | + nii.set_qform(np.diag((1.0, 2.0, 3.0, 1.0)), code=2) |
| 17 | + nii.set_sform(np.diag((1.0, 2.0, 3.0, 1.0)), code=1) |
19 | 18 | nii.to_filename(str(fname1)) |
20 | 19 |
|
21 | | - nii.set_data_dtype('float32') |
| 20 | + nii.set_data_dtype("float32") |
22 | 21 | nii.set_qform(np.eye(4), code=1) |
23 | 22 | nii.to_filename(str(fname2)) |
24 | 23 |
|
25 | | - copied = nb.load( |
26 | | - copy_header(fname1, fname2, keep_dtype=keep_dtype) |
27 | | - ) |
| 24 | + copied = nb.load(copy_header(fname1, fname2, keep_dtype=keep_dtype)) |
28 | 25 | ref = nb.load(str(fname1)) |
29 | 26 | assert np.all(copied.get_qform(coded=False) == ref.get_qform(coded=False)) |
30 | 27 | assert np.all(copied.get_sform(coded=False) == ref.get_sform(coded=False)) |
|
0 commit comments