code: def test_channel_shape(self):
      with tempfile.TemporaryDirectory() as tempdir:
          for c in (0, 1, 2, 3):
              fname = os.path.join(tempdir, f"testing{c}.nii")
              itk_writer = ITKWriter()
              itk_writer.set_data_array(
                  torch.zeros(1, 2, 3, 4),
                  channel_dim=c,
                  squeeze_end_dims=False,
              )
              itk_writer.set_metadata({})
              itk_writer.write(fname)
              itk_obj = itk.imread(fname)
              s = [1, 2, 3, 4]
              s.pop(c)
              np.testing.assert_allclose(itk.size(itk_obj), s)
  
file_path: MONAI/tests/test_itk_writer.py
imports: from __future__ import annotations
  import os
  import tempfile
  import unittest
  import numpy
  import torch
  from monai.data import ITKWriter
  from monai.utils import optional_import
label_desc: Use image node_id 1 for calling the TestITKWriter obj's underlying member method code with example usage: obj.test_channel_shape() without return types