Skip to content

Commit 3bdb6a1

Browse files
author
Ben Cipollini
committed
TST: pprint(dict) for docstring.
1 parent 6c785dd commit 3bdb6a1

File tree

3 files changed

+35
-14
lines changed

3 files changed

+35
-14
lines changed

nipype/interfaces/ants/registration.py

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,7 @@ class Registration(ANTSCommand):
380380
"""
381381
Examples
382382
--------
383-
>>> import copy
383+
>>> import copy, pprint
384384
>>> from nipype.interfaces.ants import Registration
385385
>>> reg = Registration()
386386
>>> reg.inputs.fixed_image = 'fixed1.nii'
@@ -443,15 +443,32 @@ class Registration(ANTSCommand):
443443
>>> reg4.inputs.initialize_transforms_per_stage = True
444444
>>> reg4.inputs.collapse_output_transforms = True
445445
>>> outputs = reg4._list_outputs()
446-
>>> outputs # doctest: +ELLIPSIS
447-
{'reverse_invert_flags': [], 'inverse_composite_transform': '.../nipype/testing/data/output_InverseComposite.h5', 'warped_image': '.../nipype/testing/data/output_warped_image.nii.gz', 'inverse_warped_image': <undefined>, 'forward_invert_flags': [], 'reverse_transforms': [], 'save_state': <undefined>, 'composite_transform': '.../nipype/testing/data/output_Composite.h5', 'forward_transforms': []}
446+
>>> pprint.pprint(outputs) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
447+
{'composite_transform': '.../nipype/testing/data/output_Composite.h5',
448+
'forward_invert_flags': [],
449+
'forward_transforms': [],
450+
'inverse_composite_transform': '.../nipype/testing/data/output_InverseComposite.h5',
451+
'inverse_warped_image': <undefined>,
452+
'reverse_invert_flags': [],
453+
'reverse_transforms': [],
454+
'save_state': <undefined>,
455+
'warped_image': '.../nipype/testing/data/output_warped_image.nii.gz'}
448456
449457
>>> # Test collapse transforms flag
450458
>>> reg4b = copy.deepcopy(reg4)
451459
>>> reg4b.inputs.write_composite_transform = False
452460
>>> outputs = reg4b._list_outputs()
453-
>>> outputs # doctest: +ELLIPSIS
454-
{'reverse_invert_flags': [True, False], 'inverse_composite_transform': <undefined>, 'warped_image': '.../nipype/testing/data/output_warped_image.nii.gz', 'inverse_warped_image': <undefined>, 'forward_invert_flags': [False, False], 'reverse_transforms': ['.../nipype/testing/data/output_0GenericAffine.mat', '.../nipype/testing/data/output_1InverseWarp.nii.gz'], 'save_state': <undefined>, 'composite_transform': <undefined>, 'forward_transforms': ['.../nipype/testing/data/output_0GenericAffine.mat', '.../nipype/testing/data/output_1Warp.nii.gz']}
461+
>>> pprint.pprint(outputs) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
462+
{'composite_transform': <undefined>,
463+
'forward_invert_flags': [False, False],
464+
'forward_transforms': ['.../nipype/testing/data/output_0GenericAffine.mat',
465+
'.../nipype/testing/data/output_1Warp.nii.gz'],
466+
'inverse_composite_transform': <undefined>,
467+
'inverse_warped_image': <undefined>,
468+
'reverse_invert_flags': [True, False],
469+
'reverse_transforms': ['.../nipype/testing/data/output_0GenericAffine.mat', '.../nipype/testing/data/output_1InverseWarp.nii.gz'],
470+
'save_state': <undefined>,
471+
'warped_image': '.../nipype/testing/data/output_warped_image.nii.gz'}
455472
>>> reg4b.aggregate_outputs() # doctest: +SKIP
456473
457474
>>> # Test multiple metrics per stage

nipype/interfaces/base.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1343,16 +1343,18 @@ class must be instantiated with a command argument
13431343
13441344
Examples
13451345
--------
1346-
1346+
>>> import pprint
13471347
>>> from nipype.interfaces.base import CommandLine
13481348
>>> cli = CommandLine(command='ls', environ={'DISPLAY': ':1'})
13491349
>>> cli.inputs.args = '-al'
13501350
>>> cli.cmdline
13511351
'ls -al'
13521352
1353-
>>> cli.inputs.trait_get() # doctest: +NORMALIZE_WHITESPACE
1354-
{'ignore_exception': False, 'terminal_output': 'stream',
1355-
'environ': {'DISPLAY': ':1'}, 'args': '-al'}
1353+
>>> pprint.pprint(cli.inputs.trait_get()) # doctest: +NORMALIZE_WHITESPACE
1354+
{'args': '-al',
1355+
'environ': {'DISPLAY': ':1'},
1356+
'ignore_exception': False,
1357+
'terminal_output': 'stream'}
13561358
13571359
>>> cli.inputs.get_hashval()
13581360
([('args', '-al')], '11c37f97649cd61627f4afe5136af8c0')

nipype/interfaces/io.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -917,12 +917,13 @@ class SelectFiles(IOBase):
917917
Examples
918918
--------
919919
920+
>>> import pprint
920921
>>> from nipype import SelectFiles, Node
921922
>>> templates={"T1": "{subject_id}/struct/T1.nii",
922923
... "epi": "{subject_id}/func/f[0, 1].nii"}
923924
>>> dg = Node(SelectFiles(templates), "selectfiles")
924925
>>> dg.inputs.subject_id = "subj1"
925-
>>> dg.outputs.get()
926+
>>> pprint.pprint(dg.outputs.get()) # doctest: +NORMALIZE_WHITESPACE
926927
{'T1': <undefined>, 'epi': <undefined>}
927928
928929
The same thing with dynamic grabbing of specific files:
@@ -2147,16 +2148,17 @@ class JSONFileGrabber(IOBase):
21472148
Example
21482149
-------
21492150
2151+
>>> import pprint
21502152
>>> from nipype.interfaces.io import JSONFileGrabber
21512153
>>> jsonSource = JSONFileGrabber()
21522154
>>> jsonSource.inputs.defaults = {'param1': 'overrideMe', 'param3': 1.0}
21532155
>>> res = jsonSource.run()
2154-
>>> res.outputs.get()
2155-
{'param3': 1.0, 'param1': 'overrideMe'}
2156+
>>> pprint.pprint(res.outputs.get())
2157+
{'param1': 'overrideMe', 'param3': 1.0}
21562158
>>> jsonSource.inputs.in_file = 'jsongrabber.txt'
21572159
>>> res = jsonSource.run()
2158-
>>> res.outputs.get()
2159-
{'param3': 1.0, 'param2': 4, 'param1': 'exampleStr'}
2160+
>>> pprint.pprint(res.outputs.get()) # doctest: +NORMALIZE_WHITESPACE
2161+
{'param1': u'exampleStr', 'param2': 4, 'param3': 1.0}
21602162
21612163
21622164
"""

0 commit comments

Comments
 (0)