Doing GMM classification tests in numpy