code: def test_should_explain_output(
      convolutional_model, random_data, mocker
  ):
      mocker.patch(
          "tf_explain.core.integrated_gradients.grid_display",
          side_effect=lambda x: x,
      )
      images, labels = random_data
      explainer = IntegratedGradients()
      grid = explainer.explain((images, labels), convolutional_model, 0)
  
      # Outputs is in grayscale format
      assert grid.shape == images.shape[:-1]
  
imports: import numpy
  from tf_explain.core.integrated_gradients import IntegratedGradients
label_desc: Use image node_id 1 for calling a global function with example usage: test_should_explain_output(convolutional_model, random_data, mocker) without return types