コード例 #1
0
def test_JointFusion_cmd():
    at = JointFusion()
    at.inputs.dimension = 3
    at.inputs.modalities = 1
    at.inputs.method = 'Joint[0.1,2]'
    at.inputs.output_label_image = 'fusion_labelimage_output.nii'
    warped_intensity_images = [example_data('im1.nii'),
                               example_data('im2.nii')]
    at.inputs.warped_intensity_images = warped_intensity_images
    segmentation_images = [example_data('segmentation0.nii.gz'),
                           example_data('segmentation1.nii.gz')]
    at.inputs.warped_label_images = segmentation_images
    T1_image = example_data('T1.nii')
    at.inputs.target_image = T1_image
    at.inputs.patch_radius = [3, 2, 1]
    at.inputs.search_radius = [1, 2, 3]
    expected_command = ('jointfusion 3 1 -m Joint[0.1,2] -rp 3x2x1 -rs 1x2x3'
                        ' -tg %s -g %s -g %s -l %s -l %s'
                        ' fusion_labelimage_output.nii') % (T1_image,
                                                            warped_intensity_images[0],
                                                            warped_intensity_images[1],
                                                            segmentation_images[0],
                                                            segmentation_images[1])
    assert at.cmdline == expected_command
    # setting intensity or labels with unequal lengths raises error
    with pytest.raises(AssertionError):
        at._format_arg('warped_intensity_images', InputMultiPath, warped_intensity_images + [example_data('im3.nii')])
コード例 #2
0
def test_JointFusion_cmd():
    at = JointFusion()
    at.inputs.dimension = 3
    at.inputs.modalities = 1
    at.inputs.method = 'Joint[0.1,2]'
    at.inputs.output_label_image = 'fusion_labelimage_output.nii'
    warped_intensity_images = [
        example_data('im1.nii'),
        example_data('im2.nii')
    ]
    at.inputs.warped_intensity_images = warped_intensity_images
    segmentation_images = [
        example_data('segmentation0.nii.gz'),
        example_data('segmentation1.nii.gz')
    ]
    at.inputs.warped_label_images = segmentation_images
    T1_image = example_data('T1.nii')
    at.inputs.target_image = T1_image
    at.inputs.patch_radius = [3, 2, 1]
    at.inputs.search_radius = [1, 2, 3]
    expected_command = ('jointfusion 3 1 -m Joint[0.1,2] -rp 3x2x1 -rs 1x2x3'
                        ' -tg %s -g %s -g %s -l %s -l %s'
                        ' fusion_labelimage_output.nii') % (
                            T1_image, warped_intensity_images[0],
                            warped_intensity_images[1], segmentation_images[0],
                            segmentation_images[1])
    assert at.cmdline == expected_command
    # setting intensity or labels with unequal lengths raises error
    with pytest.raises(AssertionError):
        at._format_arg('warped_intensity_images', InputMultiPath,
                       warped_intensity_images + [example_data('im3.nii')])
コード例 #3
0
def test_JointFusion_radius():
    at = JointFusion()
    set_radius = lambda attr,x,y,z: setattr(at.inputs, attr, [x, y, z])
    for attr in ['patch_radius', 'search_radius']:
        for x in range(5):
            set_radius(attr, x, x + 1, x**x)
            yield assert_equal, at._format_arg(attr, None, getattr(at.inputs, attr))[4:], '{0}x{1}x{2}'.format(x, x + 1, x**x)
コード例 #4
0
def test_JointFusion_radius():
    at = JointFusion()
    set_radius = lambda attr,x,y,z: setattr(at.inputs, attr, [x, y, z])
    for attr in ['patch_radius', 'search_radius']:
        for x in range(5):
            set_radius(attr, x, x + 1, x**x)
            yield assert_equal, at._format_arg(attr, None, getattr(at.inputs, attr))[4:], '{0}x{1}x{2}'.format(x, x + 1, x**x)
コード例 #5
0
def test_JointFusion_radius(attr, x):
    at = JointFusion()
    setattr(at.inputs, attr, [x, x + 1, x**x])
    assert at._format_arg(attr, None, getattr(
        at.inputs, attr))[4:] == "{0}x{1}x{2}".format(x, x + 1, x**x)
コード例 #6
0
def test_JointFusion_radius(attr, x):
    at = JointFusion()
    setattr(at.inputs, attr, [x, x+1, x**x])
    assert at._format_arg(attr, None, getattr(at.inputs, attr))[4:] == '{0}x{1}x{2}'.format(x, x + 1, x**x)