code: def test_should_transform_to_uint_255_image(
      input_image, expected_output
  ):
      output = image_to_uint_255(input_image)
  
      np.testing.assert_almost_equal(output, expected_output)
  
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 5 for calling a global function with example usage: test_should_transform_to_uint_255_image(input_image, expected_output) without return types