code: def test_get_img_shape_on_3d_image():
      n = 5
      channels = 4
      dim1 = 1
      dim2 = 2
      dim3 = 3
  
      K.set_image_data_format("channels_first")
      assert (n, channels, dim1, dim2, dim3) == utils.get_img_shape(
          K.ones(shape=(n, channels, dim1, dim2, dim3))
      )
  
      K.set_image_data_format("channels_last")
      assert (n, channels, dim1, dim2, dim3) == utils.get_img_shape(
          K.ones(shape=(n, dim1, dim2, dim3, channels))
      )
  
imports: import pytest
  from vis.utils import utils
  from keras import backend
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_img_shape_on_3d_image() without return types