# HG changeset patch
# User Amine Sehili <amine.sehili@gmail.com>
# Date 1550518930 -3600
# Node ID 5ff29f28637404732da07916571c98832ad4e532
# Parent  d42c0c2858c55c39606b26388ed3f26da2db4cd2
Add tests for _guess_audio_format

diff -r d42c0c2858c5 -r 5ff29f286374 tests/test_io.py
--- 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),