Add more tests for clustering math