changeset 289:69ca1c64a9b0

Return 'wav' when 'wave' is guessed in _guess_audio_format

--- a/auditok/io.py	Fri Oct 04 22:02:37 2019 +0100
+++ b/auditok/io.py	Sat Oct 05 14:26:03 2019 +0200
@@ -95,8 +95,14 @@
 def _guess_audio_format(fmt, filename):
     if fmt is None:
         extension = os.path.splitext(filename.lower())[1][1:]
-        return extension if extension else None
-    return fmt.lower()
+        if extension:
+            fmt = extension
+        else:
+            return None
+    fmt == fmt.lower()
+    if fmt == "wave":
+        fmt = "wav"
+    return fmt
 
 
 def _normalize_use_channel(use_channel):

--- a/tests/test_io.py	Fri Oct 04 22:02:37 2019 +0100
+++ b/tests/test_io.py	Sat Oct 05 14:26:03 2019 +0200
@@ -67,6 +67,8 @@
         extention_no_format=(None, "filename.wav", "wav"),
         format_no_extension=("wav", "filename", "wav"),
         no_format_no_extension=(None, "filename", None),
+        wave_as_wav=("wave", "filename", "wav"),
+        wave_as_wav_extension=(None, "filename.wave", "wav"),
     )
     def test_guess_audio_format(self, fmt, filename, expected):
         result = _guess_audio_format(fmt, filename)