code: def test_num_axes(image_array_2bands):
      """Test the number of axes.
  
      If provided with a 2 band array, plot_bands should return 3 axes.
      And 2 colorbars
      """
      ax = ep.plot_bands(image_array_2bands)
      ax = list(ax)
      cb = [a.images[0].colorbar for a in ax if a.images]
  
      assert len(ax) == 3
      assert len(cb) == 2
      plt.close()
  
imports: import pytest
  import matplotlib.pyplot
  import matplotlib.colors
  import earthpy.plot
  import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: test_num_axes(image_array_2bands) without return types