Update the kmeans tests to use the new dataset structure.