code: def test_correct_shapes_gram():
      assert orthogonal_mp_gram(
          G, Xy[:, 0], n_nonzero_coefs=5
      ).shape == (n_features,)
      assert orthogonal_mp_gram(G, Xy, n_nonzero_coefs=5).shape == (
          n_features,
          3,
      )
  
imports: import numpy
  import pytest
  from sklearn.datasets import make_sparse_coded_signal
  from sklearn.linear_model import LinearRegression, OrthogonalMatchingPursuit, OrthogonalMatchingPursuitCV, orthogonal_mp, orthogonal_mp_gram
  from sklearn.utils import check_random_state
  from sklearn.utils._testing import assert_allclose, assert_array_almost_equal, assert_array_equal, ignore_warnings
label_desc: Use image node_id 2 for calling a global function with example usage: test_correct_shapes_gram() without return types