code: def test_should_return_a_grid_square_by_default(
      grid_display_kwargs, expected_shape, array_to_display
  ):
      grid = grid_display(array_to_display, **grid_display_kwargs)
  
      assert grid.shape == expected_shape
  
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 2 for calling a global function with example usage: test_should_return_a_grid_square_by_default(grid_display_kwargs, expected_shape, array_to_display) without return types