def METHOD_NAME(kernel_embedding, x):
    """Test output shape of kernel mean."""
    kernel_mean_shape = (np.atleast_2d(x).shape[0],)
    assert kernel_embedding.kernel_mean(x).shape == kernel_mean_shape, (
        f"Kernel mean of {type(kernel_embedding)} has shape"
        f" {kernel_embedding.kernel_mean(x).shape} instead of {kernel_mean_shape}"
    )