code: def test_should_display_heatmap(mocker):
      mock_add_weighted = mocker.patch(
          "tf_explain.utils.display.cv2.addWeighted"
      )
      mock_apply_color_map = mocker.patch(
          "tf_explain.utils.display.cv2.applyColorMap"
      )
      mock_cvt_color = mocker.patch(
          "tf_explain.utils.display.cv2.cvtColor",
          return_value=mocker.sentinel.heatmap,
      )
  
      heatmap = np.random.random((3, 3))
      original_image = np.zeros((10, 10, 3))
  
      output = heatmap_display(heatmap, original_image)
  
      assert output == mocker.sentinel.heatmap
      assert mock_add_weighted.call_count == 1
      assert mock_apply_color_map.call_count == 1
      assert mock_cvt_color.call_count == 3
  
file_path: tf-explain/tests/utils/test_display.py
imports: import numpy
  import pytest
  from tf_explain.utils.display import grid_display, heatmap_display, image_to_uint_255
label_desc: Use image node_id 6 for calling a global function with example usage: test_should_display_heatmap(mocker) without return types