Skip to content

Commit 7f03b0a

Browse files
committed
TST: Check that the general loader can load a PET HDF5 dataset
Check that the general loader can load a PET HDF5 dataset.
1 parent 5b7fb3a commit 7f03b0a

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test/test_data_pet.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
import pytest
3030
from nitransforms.linear import Affine
3131

32+
from nifreeze.data import load
3233
from nifreeze.data.pet import PET, _compute_frame_duration, _compute_uptake_statistic, from_nii
3334

3435

@@ -62,6 +63,12 @@ def random_nifti_file(tmp_path, setup_random_uniform_spatial_data) -> Path:
6263
return _filename
6364

6465

66+
def test_main(datadir):
67+
input_file = datadir / "pet.h5"
68+
69+
assert isinstance(load(input_file), PET)
70+
71+
6572
@pytest.mark.parametrize(
6673
"midframe, expected",
6774
[

0 commit comments

Comments
 (0)