|
| 1 | +# AUTO-GENERATED by tools/checkspecs.py - DO NOT EDIT |
| 2 | +from ..segmentation import JointFusion |
| 3 | + |
| 4 | + |
| 5 | +def test_JointFusion_inputs(): |
| 6 | + input_map = dict( |
| 7 | + alpha=dict(argstr="-a %s", usedefault=True,), |
| 8 | + args=dict(argstr="%s",), |
| 9 | + atlas_image=dict(argstr="-g %s...", mandatory=True,), |
| 10 | + atlas_segmentation_image=dict(argstr="-l %s...", mandatory=True,), |
| 11 | + beta=dict(argstr="-b %s", usedefault=True,), |
| 12 | + constrain_nonnegative=dict(argstr="-c", usedefault=True,), |
| 13 | + dimension=dict(argstr="-d %d",), |
| 14 | + environ=dict(nohash=True, usedefault=True,), |
| 15 | + exclusion_image=dict(), |
| 16 | + exclusion_image_label=dict(argstr="-e %s", requires=["exclusion_image"],), |
| 17 | + mask_image=dict(argstr="-x %s", extensions=None,), |
| 18 | + num_threads=dict(nohash=True, usedefault=True,), |
| 19 | + out_atlas_voting_weight_name_format=dict( |
| 20 | + requires=[ |
| 21 | + "out_label_fusion", |
| 22 | + "out_intensity_fusion_name_format", |
| 23 | + "out_label_post_prob_name_format", |
| 24 | + ], |
| 25 | + ), |
| 26 | + out_intensity_fusion_name_format=dict(argstr="",), |
| 27 | + out_label_fusion=dict(argstr="%s", extensions=None, hash_files=False,), |
| 28 | + out_label_post_prob_name_format=dict( |
| 29 | + requires=["out_label_fusion", "out_intensity_fusion_name_format"], |
| 30 | + ), |
| 31 | + patch_metric=dict(argstr="-m %s",), |
| 32 | + patch_radius=dict(argstr="-p %s", maxlen=3, minlen=3,), |
| 33 | + retain_atlas_voting_images=dict(argstr="-f", usedefault=True,), |
| 34 | + retain_label_posterior_images=dict( |
| 35 | + argstr="-r", requires=["atlas_segmentation_image"], usedefault=True, |
| 36 | + ), |
| 37 | + search_radius=dict(argstr="-s %s", usedefault=True,), |
| 38 | + target_image=dict(argstr="-t %s", mandatory=True,), |
| 39 | + verbose=dict(argstr="-v",), |
| 40 | + ) |
| 41 | + inputs = JointFusion.input_spec() |
| 42 | + |
| 43 | + for key, metadata in list(input_map.items()): |
| 44 | + for metakey, value in list(metadata.items()): |
| 45 | + assert getattr(inputs.traits()[key], metakey) == value |
| 46 | + |
| 47 | + |
| 48 | +def test_JointFusion_outputs(): |
| 49 | + output_map = dict( |
| 50 | + out_atlas_voting_weight=dict(), |
| 51 | + out_intensity_fusion=dict(), |
| 52 | + out_label_fusion=dict(extensions=None,), |
| 53 | + out_label_post_prob=dict(), |
| 54 | + ) |
| 55 | + outputs = JointFusion.output_spec() |
| 56 | + |
| 57 | + for key, metadata in list(output_map.items()): |
| 58 | + for metakey, value in list(metadata.items()): |
| 59 | + assert getattr(outputs.traits()[key], metakey) == value |
0 commit comments