def METHOD_NAME():
    """Test that GMM estimation is successful when the correct covariance type
    is passed in as a keyword argument.
    """
    gmm = learn_gmm(
        np.random.random((10, 10)), n_modes=2,
        gm_args={'covariance_type': 'diag'}
    )
    assert gmm.means_ is not None
    assert gmm.covariances_ is not None
    assert gmm.weights_ is not None