code: def _get_y_shapes(y_dtype, shape, rowvar):
      # Helper function for testCov.
      if y_dtype is None:
          return [None]
      if len(shape) == 1:
          return [shape]
      elif rowvar or shape[0] == 1:
          return [(1, shape[-1]), (2, shape[-1]), (5, shape[-1])]
      return [(shape[0], 1), (shape[0], 2), (shape[0], 5)]
  
imports: import collections
  from functools import partial
  import itertools
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import jax
  from jax import numpy
  from jax._src import config
  from jax._src import dtypes
  from jax._src import test_util
  from jax._src.util import NumpyComplexWarning
label_desc: Use image node_id 4 for calling a global function with example usage: _get_y_shapes(y_dtype, shape, rowvar) and returns: list, list, list, list