code: def test_basic_functionality(self):
      heatmaps_arr = np.float32(
          [
              [0.5, 0.0, 0.0, 0.5],
              [0.0, 1.0, 1.0, 0.0],
              [0.0, 1.0, 1.0, 0.0],
              [0.5, 0.0, 0.0, 0.5],
          ]
      )
      heatmaps = ia.HeatmapsOnImage(heatmaps_arr, shape=(4, 4, 3))
  
      heatmaps_drawn = heatmaps.draw()[0]
      assert heatmaps_drawn.shape == (4, 4, 3)
      v1 = heatmaps_drawn[0, 1]
      v2 = heatmaps_drawn[0, 0]
      v3 = heatmaps_drawn[1, 1]
  
      v1_coords = [
          (0, 1),
          (0, 2),
          (1, 0),
          (1, 3),
          (2, 0),
          (2, 3),
          (3, 1),
          (3, 2),
      ]
      v2_coords = [(0, 0), (0, 3), (3, 0), (3, 3)]
      v3_coords = [(1, 1), (1, 2), (2, 1), (2, 2)]
  
      for y, x in v1_coords:
          assert np.allclose(heatmaps_drawn[y, x], v1)
  
      for y, x in v2_coords:
          assert np.allclose(heatmaps_drawn[y, x], v2)
  
      for y, x in v3_coords:
          assert np.allclose(heatmaps_drawn[y, x], v3)
  
imports: from __future__ import print_function, division, absolute_import
  import sys
  import numpy
  import six.moves
  import imgaug
label_desc: Use image node_id 1 for calling the TestHeatmapsOnImage_draw obj's underlying member method code with example usage: obj.test_basic_functionality() without return types