def test_identity_mean():
    """Test the logdet mean"""
    covmats = generate_cov(100, 3)
    C = mean_identity(covmats)
    assert_array_equal(C, np.eye(3))
def test_identity_mean():
    """Test the logdet mean"""
    covmats, diags, A = generate_cov(100, 3)
    C = mean_identity(covmats)
    assert_array_equal(C, np.eye(3))
def test_identity_mean(get_covmats):
    """Test the identity mean"""
    n_matrices, n_channels = 100, 3
    covmats = get_covmats(n_matrices, n_channels)
    C = mean_identity(covmats)
    assert np.all(C == np.eye(n_channels))