code: def test_zero_flows():
      """Apply _check_zero_flow_correctness() for a few sizes and types."""
      shapes_to_try = [[3, 4, 5, 6], [1, 2, 2, 1]]
      for shape in shapes_to_try:
          _check_zero_flow_correctness(
              shape, image_type="float32", flow_type="float32"
          )
  
file_path: addons/tensorflow_addons/image/tests/dense_image_warp_test.py
imports: import pytest
  import math
  import numpy
  import tensorflow
  from tensorflow_addons.image import dense_image_warp
  from tensorflow_addons.image import interpolate_bilinear
label_desc: Use image node_id 6 for calling a global function with example usage: test_zero_flows() without return types