code: def test_colorbar_height(basic_image):
      """Test that the colorbar ax height matches the image axis height."""
      f, ax = plt.subplots(figsize=(5, 5))
      im = ax.imshow(basic_image, cmap="RdYlGn")
      cb = ep.colorbar(im)
      assert (
          cb.ax.get_position().height == im.axes.get_position().height
      )
      plt.close(f)
  
imports: import pytest
  import matplotlib.pyplot
  import earthpy.plot
label_desc: Use image node_id 1 for calling a global function with example usage: test_colorbar_height(basic_image) without return types