code: def test_get_grid_locations():
      image_height = 5
      image_width = 3
      grid = _get_grid_locations(image_height, image_width)
      for i in range(image_height):
          for j in range(image_width):
              assert grid[i, j, 0] == i
              assert grid[i, j, 1] == j
  
file_path: addons/tensorflow_addons/image/tests/sparse_image_warp_test.py
imports: from collections import namedtuple
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.image import sparse_image_warp
  from tensorflow_addons.image.sparse_image_warp import _get_boundary_locations
  from tensorflow_addons.image.sparse_image_warp import _get_grid_locations
  from tensorflow_addons.utils.resource_loader import get_path_to_datafile
label_desc: Use image node_id 4 for calling a global function with example usage: test_get_grid_locations() without return types