ea1bdc71e554847b6b982058547e0b26bbd20faf,nilearn/plotting/tests/test_matrix_plotting.py,,test_matrix_plotting,#,12

Before Change

    plt.close()
    // test if reordering with default linkage works
    ax = plot_matrix(mat, labels=labels, reorder=True)
    plt.close()

After Change

    ax.axes.set_title("Title")
    plt.close()
    import scipy
    if int(scipy.__version__.split(".")[0]) >= 1:
        // test if reordering with default linkage works
        idx = [2, 3, 5]
        from itertools import permutations
        // make symmetric matrix of similarities so we can get a block
        for perm in permutations(idx, 2):
            mat[perm] = 1
        ax = plot_matrix(mat, labels=labels, reorder=True)
        reordered_labels = [int(lbl.get_text())
                            for lbl in ax.axes.get_xticklabels()]
        // block order does not matter
        assert_true(reordered_labels[:3] == idx or reordered_labels[-3:] == idx,
                    "Clustering does not find block structure.")
        plt.close()

Instances