Updated tests for python_speech_features

Signed-off-by: Vitaly Lavrukhin <[email protected]>

Author: vsl9

open_seq2seq/data/speech2text/speech_utils.py

@@ -460,10 +460,10 @@ def get_speech_features_psf(signal, sample_freq, num_features,
     audio_duration (float): duration of the signal in seconds
   """
   if augmentation is not None:
-    signal = augment_audio_signal(signal, sample_freq, augmentation)
-  else:
-    signal = (normalize_signal(signal.astype(np.float32)) * 32767.0).astype(
-        np.int16)
+    signal = augment_audio_signal(signal.astype(np.float32), 
+        sample_freq, augmentation)
+  signal = (normalize_signal(signal.astype(np.float32)) * 32767.0).astype(
+      np.int16)
 
   audio_duration = len(signal) * 1.0 / sample_freq
 

open_seq2seq/data/speech2text/speech_utils_test.py

@@ -20,6 +20,7 @@ class SpeechUtilsTests(tf.test.TestCase):
   def test_augment_audio_signal(self):
     filename = 'open_seq2seq/test_utils/toy_speech_data/wav_files/46gc040q.wav'
     freq_s, signal = wave.read(filename)
+    signal = signal.astype(np.float32)
     augmentation = {
         'speed_perturbation_ratio': 0.2,
         'noise_level_min': -90,