# HG changeset patch
# User Amine Sehili <amine.sehili@gmail.com>
# Date 1549655496 -3600
# Node ID 7c5f325dae7aec90a4d941f541654199b0ff2771
# Parent  fb31a72504027e97f7d59772f6313234887a8a75
Add tests for _load_wave with channel mixing

diff -r fb31a7250402 -r 7c5f325dae7a tests/test_io.py
--- a/tests/test_io.py	Thu Feb 07 20:26:33 2019 +0100
+++ b/tests/test_io.py	Fri Feb 08 20:51:36 2019 +0100
@@ -468,6 +468,31 @@
         self.assertEqual(data, expected)
 
     @genty_dataset(
+        mono=("mono_400Hz", (400,)),
+        three_channel=("3channel_400-800-1600Hz", (400, 800, 1600)),
+    )
+    def test_load_wave_mix(self, filename_suffix, frequencies):
+        sampling_rate = 16000
+        sample_width = 2
+        channels = len(frequencies)
+        mono_channels = [PURE_TONE_DICT[freq] for freq in frequencies]
+        fmt = DATA_FORMAT[sample_width]
+        expected = _array_to_bytes(
+            array(
+                fmt,
+                (sum(samples) // channels for samples in zip(*mono_channels)),
+            )
+        )
+        filename = "tests/data/test_16KHZ_{}.wav".format(filename_suffix)
+        audio_source = _load_wave(filename, use_channel="mix")
+        mixed = audio_source._buffer
+        self.assertEqual(mixed, expected)
+        self.assertIsInstance(audio_source, BufferAudioSource)
+        self.assertEqual(audio_source.sampling_rate, sampling_rate)
+        self.assertEqual(audio_source.sample_width, sample_width)
+        self.assertEqual(audio_source.channels, 1)
+
+    @genty_dataset(
         mono=("mono_400Hz.wav", (400,)),
         three_channel=("3channel_400-800-1600Hz.wav", (400, 800, 1600)),
     )