def test_save_load():
    '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)
    save_load_compare(STFT(input_shape=input_shape, pad_begin=True), batch_src, allclose_complex_numbers)
    save_load_compare(get_melspectrogram_layer(input_shape=input_shape, return_decibel=True), batch_src, np.testing.assert_allclose)
    save_load_compare(get_log_frequency_spectrogram_layer(input_shape=input_shape, return_decibel=True), batch_src, np.testing.assert_allclose)
    save_load_compare(get_stft_mag_phase(input_shape=input_shape, return_decibel=True), batch_src, np.testing.assert_allclose)
    save_load_compare(get_stft_magnitude_layer(input_shape=input_shape), batch_src, np.testing.assert_allclose)
test saving/loading of models that has stft, melspectorgrma, and log frequency.
def test_save_load():
    
    (src_mono, batch_src, input_shape) = get_audio(data_format='channels_last', n_ch=1)
    save_load_compare(STFT(input_shape=input_shape, pad_begin=True), batch_src, allclose_complex_numbers)
    save_load_compare(get_melspectrogram_layer(input_shape=input_shape, return_decibel=True), batch_src, np.testing.assert_allclose)
    save_load_compare(get_log_frequency_spectrogram_layer(input_shape=input_shape, return_decibel=True), batch_src, np.testing.assert_allclose)
    save_load_compare(get_stft_mag_phase(input_shape=input_shape, return_decibel=True), batch_src, np.testing.assert_allclose)
    save_load_compare(get_stft_magnitude_layer(input_shape=input_shape), batch_src, np.testing.assert_allclose)