# HG changeset patch
# User Amine Sehili <amine.sehili@gmail.com>
# Date 1549912786 -3600
# Node ID 0125bd4ee31779b7d880d3d3d85609c0f807a6f9
# Parent  d8ae412e5aa8fe31afd962e9de26875e4e0e098d
Add tests for to_file with wave data

diff -r d8ae412e5aa8 -r 0125bd4ee317 tests/test_io.py
--- a/tests/test_io.py	Sun Feb 10 21:21:19 2019 +0100
+++ b/tests/test_io.py	Mon Feb 11 20:19:46 2019 +0100
@@ -580,3 +580,22 @@
         _save_raw(filename, data)
         self.assertTrue(filecmp.cmp(filename, exp_filename, shallow=False))
         tmpdir.cleanup()
+
+    @genty_dataset(
+        wav_with_audio_format=("audio", "wav"),
+        wav_with_extension=("audio.wav", None),
+        wav_with_audio_format_and_extension=("audio.mp3", "wav"),
+        wave_with_audio_format=("audio", "wave"),
+        wave_with_extension=("audio.wave", None),
+        wave_with_audio_format_and_extension=("audio.mp3", "wave"),
+    )
+    def test_to_file_wav(self, filename, audio_format):
+        exp_filename = "tests/data/test_16KHZ_mono_400Hz.wav"
+        tmpdir = TemporaryDirectory()
+        filename = os.path.join(tmpdir.name, filename)
+        data = _array_to_bytes(PURE_TONE_DICT[400])
+        _save_wave(
+            filename, data, sampling_rate=16000, sample_width=2, channels=1
+        )
+        self.assertTrue(filecmp.cmp(filename, exp_filename, shallow=False))
+        tmpdir.cleanup()