code: def test_should_produce_heatmap(
      convolutional_model, random_data, mocker
  ):
      mocker.patch(
          "tf_explain.core.occlusion_sensitivity.grid_display",
          return_value=mocker.sentinel.grid,
      )
  
      explainer = OcclusionSensitivity()
      grid = explainer.explain(random_data, convolutional_model, 0, 10)
  
      assert grid == mocker.sentinel.grid
  
imports: import math
  import numpy
  from tf_explain.core.occlusion_sensitivity import OcclusionSensitivity
label_desc: Use image node_id 2 for calling a global function with example usage: test_should_produce_heatmap(convolutional_model, random_data, mocker) without return types