code: def test__ensure_shape():
      img, label = tf.random.normal((32, 32, 3)), 0
      img1, label1 = clf._ensure_shape(img, label)
      assert np.all(img1 == img)
      assert np.all(label1 == label)
  
imports: import numpy
  import tensorflow
  from chitra.dataloader import Clf
label_desc: Use image node_id 2 for calling a global function with example usage: test__ensure_shape() without return types