e8aff8a890da8670ea8af8783caf3984c4dde558,dipy/align/tests/test_transforms.py,,test_number_of_parameters,#,14

Before Change

    actual_2d = []
    actual_3d = []
    for transform in transforms:
        t = transform_type[transform]
        actual_2d.append(number_of_parameters(t, 2))
        actual_3d.append(number_of_parameters(t, 3))

    assert_array_equal(actual_2d, expected_2d)

After Change

                           assert_raises)

def test_number_of_parameters():
    expected_params = {("TRANSLATION", 2) : 2,
                       ("TRANSLATION", 3) : 3,
                       ("ROTATION", 2) : 1,
                       ("ROTATION", 3) : 3,
                       ("RIGID", 2) : 3,
                       ("RIGID", 3) : 6,
                       ("SCALING", 2) : 1,
                       ("SCALING", 3) : 1,
                       ("AFFINE", 2) : 6,
                       ("AFFINE", 3) : 12}

    for ttype, transform in regtransforms.iteritems():
        assert_equal(transform.get_number_of_parameters(), expected_params[ttype])

Instances