def METHOD_NAME(u):
    dist_universe = diffusionmap.DistanceMatrix(u, select='backbone').run()
    ag = u.select_atoms('protein')
    dist_ag = diffusionmap.DistanceMatrix(ag, select='backbone').run()
    assert_allclose(dist_universe.results.dist_matrix,
                    dist_ag.results.dist_matrix)