code: def test_save_load_channel_swap(data_format, save_format):
      src_mono, batch_src, input_shape = get_audio(
          data_format="channels_last", n_ch=1
      )
  
      save_load_compare(
          ChannelSwap(input_shape=input_shape),
          batch_src,
          np.testing.assert_allclose,
          save_format=save_format,
          layer_class=ChannelSwap,
          training=None,
      )
  
file_path: kapre/tests/test_augmentation.py
imports: import pytest
  import numpy
  import tensorflow
  from kapre.augmentation import ChannelSwap, SpecAugment
  from kapre.backend import _CH_FIRST_STR, _CH_LAST_STR, _CH_DEFAULT_STR
  from utils import get_audio, get_spectrogram, save_load_compare
label_desc: Use image node_id 5 for calling a global function with example usage: test_save_load_channel_swap(data_format, save_format) without return types