code: def test_nchw(self):
      x = self.ndimarange((2, 3, 4, 5))
      self.assertEqual(
          objax.util.image.nchw(x).tolist(),
          x.transpose((0, 3, 1, 2)).tolist(),
      )
      self.assertEqual(
          objax.util.image.nchw(jn.array(x)).tolist(),
          x.transpose((0, 3, 1, 2)).tolist(),
      )
      x = self.ndimarange((2, 3, 4, 5, 6))
      self.assertEqual(
          objax.util.image.nchw(x).tolist(),
          x.transpose((0, 1, 4, 2, 3)).tolist(),
      )
      self.assertEqual(
          objax.util.image.nchw(jn.array(x)).tolist(),
          x.transpose((0, 1, 4, 2, 3)).tolist(),
      )
  
file_path: objax/tests/util_image.py
imports: import io
  import tempfile
  import unittest
  from typing import Tuple
  import jax.numpy
  import numpy
  from PIL import Image
  import objax
label_desc: Use image node_id 2 for calling the TestUtilImage obj's underlying member method code with example usage: obj.test_nchw() without return types