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