code: def test_save_load_spec_augment(data_format, save_format):
      batch_src, input_shape = get_spectrogram(data_format=data_format)
  
      spec_augment = SpecAugment(
          input_shape=input_shape,
          freq_mask_param=5,
          time_mask_param=10,
          n_freq_masks=4,
          n_time_masks=3,
          mask_value=0.0,
          data_format=data_format,
      )
      save_load_compare(
          spec_augment,
          batch_src,
          np.testing.assert_allclose,
          save_format=save_format,
          layer_class=SpecAugment,
          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 6 for calling a global function with example usage: test_save_load_spec_augment(data_format, save_format) without return types