Add tests for gmm