code: def test_3d_uint8_array(self):
      hm = ia.HeatmapsOnImage.from_uint8(
          np.uint8([[0, 128, 255], [255, 128, 0]])[..., np.newaxis],
          (20, 30, 3),
      )
      assert hm.shape == (20, 30, 3)
      assert hm.arr_0to1.shape == (2, 3, 1)
      assert np.allclose(
          hm.arr_0to1[..., 0],
          np.float32([[0, 128 / 255, 1.0], [1.0, 128 / 255, 0]]),
      )
  
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_from_uint8 obj's underlying member method code with example usage: obj.test_3d_uint8_array() without return types