code: def test_save_load(save_format):
      """test saving/loading of models that has stft, melspectorgrma, and log frequency."""
  
      src_mono, batch_src, input_shape = get_audio(
          data_format="channels_last", n_ch=1
      )
      # test STFT save/load
      save_load_compare(
          STFT(input_shape=input_shape, pad_begin=True),
          batch_src,
          allclose_complex_numbers,
          save_format,
          STFT,
      )
  
      # test ConcatenateFrequencyMap
      specs_batch = np.random.randn(2, 3, 5, 4).astype(np.float32)
      save_load_compare(
          ConcatenateFrequencyMap(input_shape=specs_batch.shape[1:]),
          specs_batch,
          np.testing.assert_allclose,
          save_format,
          ConcatenateFrequencyMap,
      )
  
      if save_format == "tf":
          # test melspectrogram save/load
          save_load_compare(
              get_melspectrogram_layer(
                  input_shape=input_shape, return_decibel=True
              ),
              batch_src,
              np.testing.assert_allclose,
              save_format,
          )
          # test log frequency spectrogram save/load
          save_load_compare(
              get_log_frequency_spectrogram_layer(
                  input_shape=input_shape, return_decibel=True
              ),
              batch_src,
              np.testing.assert_allclose,
              save_format,
          )
          # test stft_mag_phase
          save_load_compare(
              get_stft_mag_phase(
                  input_shape=input_shape, return_decibel=True
              ),
              batch_src,
              np.testing.assert_allclose,
              save_format,
          )
          # test stft mag
          save_load_compare(
              get_stft_magnitude_layer(input_shape=input_shape),
              batch_src,
              np.testing.assert_allclose,
              save_format,
          )
  
imports: import pytest
  import numpy
  import tensorflow
  import tensorflow.keras
  import librosa
  from kapre import STFT, Magnitude, Phase, Delta, InverseSTFT, ApplyFilterbank, ConcatenateFrequencyMap, STFTTflite, MagnitudeTflite, PhaseTflite
  from kapre.composed import get_melspectrogram_layer, get_log_frequency_spectrogram_layer, get_stft_mag_phase, get_perfectly_reconstructing_stft_istft, get_stft_magnitude_layer, get_frequency_aware_conv2d
  from utils import get_audio, save_load_compare, predict_using_tflite
label_desc: Use image node_id 15 for calling a global function with example usage: test_save_load(save_format) without return types