code: def test_should_raise_warning_if_grid_size_is_too_small(
      array_to_display,
  ):
      with pytest.warns(Warning) as w:
          grid = grid_display(
              array_to_display, num_rows=1, num_columns=1
          )
  
      assert (
          w[0].message.args[0]
          == "Given values for num_rows and num_columns doesn't allow to display all images. Values have been overrided to respect at least num_columns"
      )
      assert grid.shape == (28 * 7, 28, 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 3 for calling a global function with example usage: test_should_raise_warning_if_grid_size_is_too_small(array_to_display) without return types