code: def test_should_fill_with_zeros_if_missing_elements(array_to_display):
      grid = grid_display(array_to_display)
  
      assert grid.shape == (28 * 3, 28 * 3, 3)
      np.testing.assert_equal(grid[56:, 28:, :], np.zeros((28, 56, 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 4 for calling a global function with example usage: test_should_fill_with_zeros_if_missing_elements(array_to_display) without return types