d4bed651658b585e78581ea8dd3abd4966a2e91d,dipy/direction/tests/test_peaks.py,,test_peaksFromModel,#,449

Before Change

    assert_array_equal(pam.peak_indices[mask, 1:], -1)

    // Test serialization and deserialization:
    b = BytesIO()
    pickle.dump(pam, b)
    b.seek(0)
    new_pam = pickle.load(b)
    b.close()

    for attr in ["peak_dirs", "peak_values", "peak_indices", "gfa", "qa",
                 "shm_coeff", "B", "odf"]:

After Change

    // Test serialization and deserialization:
    for normalize_peaks in [True, False]:
        for return_odf in [True, False]:
            for return_sh in [True, False]:
                pam = peaks_from_model(model, data, _sphere, .5, 45,
                                       normalize_peaks=normalize_peaks,
                                       return_odf=return_odf,
                                       return_sh=return_sh)

                b = BytesIO()
                pickle.dump(pam, b)
                b.seek(0)
                new_pam = pickle.load(b)
                b.close()

                for attr in ["peak_dirs", "peak_values", "peak_indices",
                             "gfa", "qa", "shm_coeff", "B", "odf"]:
                    assert_array_equal(getattr(pam, attr),
                                       getattr(new_pam, attr))
                    assert_array_equal(pam.sphere.vertices,
                                       new_pam.sphere.vertices)


def test_peaksFromModelParallel():
    SNR = 100
    S0 = 100

Instances