Mercurial > hg > auditok

changeset 143:5ff29f286374

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add tests for _guess_audio_format
author Amine Sehili <amine.sehili@gmail.com>
date Mon, 18 Feb 2019 20:42:10 +0100
parents d42c0c2858c5
children fae263792c2e
files tests/test_io.py
diffstat 1 files changed, 12 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/tests/test_io.py	Sun Feb 17 14:18:47 2019 +0100
+++ b/tests/test_io.py	Mon Feb 18 20:42:10 2019 +0100
@@ -12,6 +12,7 @@
     AudioParameterError,
     BufferAudioSource,
     check_audio_data,
+    _guess_audio_format,
     _get_audio_parameters,
     _array_to_bytes,
     _mix_audio_channels,
@@ -103,6 +104,17 @@
             self.assertIsNone(check_audio_data(data, sample_width, channels))
 
     @genty_dataset(
+        extention_and_format_same=("wav", "filename.wav", "wav"),
+        extention_and_format_different=("wav", "filename.mp3", "wav"),
+        extention_no_format=(None, "filename.wav", "wav"),
+        format_no_extension=("wav", "filename", "wav"),
+        no_format_no_extension=(None, "filename", None),
+    )
+    def test_guess_audio_format(self, fmt, filename, expected):
+        result = _guess_audio_format(fmt, filename)
+        self.assertEqual(result, expected)
+
+    @genty_dataset(
         mono_1byte=([400], 1),
         stereo_1byte=([400, 600], 1),
         three_channel_1byte=([400, 600, 2400], 1),