Improved tests for euclidean_distances_argmin