|
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