Skip to content

Commit b400dd5

Browse files
committed
TEST: Capture stdout in some GIFTI tests
1 parent cf9cf15 commit b400dd5

File tree

1 file changed

+18
-5
lines changed

1 file changed

+18
-5
lines changed

nibabel/gifti/tests/test_gifti.py

Lines changed: 18 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -404,13 +404,17 @@ def assign_rgba(gl, val):
404404
assert np.all([elem is None for elem in gl4.rgba])
405405

406406

407-
def test_print_summary():
408-
for fil in [DATA_FILE1, DATA_FILE2, DATA_FILE3, DATA_FILE4, DATA_FILE5, DATA_FILE6]:
409-
gimg = load(fil)
410-
gimg.print_summary()
407+
@pytest.mark.parametrize(
408+
'fname', [DATA_FILE1, DATA_FILE2, DATA_FILE3, DATA_FILE4, DATA_FILE5, DATA_FILE6]
409+
)
410+
def test_print_summary(fname, capsys):
411+
gimg = load(fname)
412+
gimg.print_summary()
413+
captured = capsys.readouterr()
414+
assert captured.out.startswith('----start----\n')
411415

412416

413-
def test_gifti_coord():
417+
def test_gifti_coord(capsys):
414418
from ..gifti import GiftiCoordSystem
415419

416420
gcs = GiftiCoordSystem()
@@ -419,6 +423,15 @@ def test_gifti_coord():
419423
# Smoke test
420424
gcs.xform = None
421425
gcs.print_summary()
426+
captured = capsys.readouterr()
427+
assert captured.out == '\n'.join(
428+
[
429+
'Dataspace: NIFTI_XFORM_UNKNOWN',
430+
'XFormSpace: NIFTI_XFORM_UNKNOWN',
431+
'Affine Transformation Matrix: ',
432+
' None\n',
433+
]
434+
)
422435
gcs.to_xml()
423436

424437

0 commit comments

Comments
 (0)