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

diff -r a9d74315f2ee -r d8ae412e5aa8 tests/test_io.py
--- a/tests/test_io.py	Sat Feb 09 18:42:14 2019 +0100
+++ b/tests/test_io.py	Sun Feb 10 21:21:19 2019 +0100
@@ -2,7 +2,7 @@
 import sys
 import math
 from array import array
-from tempfile import NamedTemporaryFile
+from tempfile import NamedTemporaryFile, TemporaryDirectory
 import filecmp
 from unittest import TestCase
 from genty import genty, genty_dataset
@@ -565,3 +565,18 @@
             del params[missing_param]
             srate, swidth, channels, _ = _get_audio_parameters(params)
             _save_wave("audio", b"\0\0", srate, swidth, channels)
+
+    @genty_dataset(
+        raw_with_audio_format=("audio", "raw"),
+        raw_with_extension=("audio.raw", None),
+        raw_with_audio_format_and_extension=("audio.mp3", "raw"),
+        raw_no_audio_format_nor_extension=("audio", None),
+    )
+    def test_to_file_raw(self, filename, audio_format):
+        exp_filename = "tests/data/test_16KHZ_mono_400Hz.raw"
+        tmpdir = TemporaryDirectory()
+        filename = os.path.join(tmpdir.name, filename)
+        data = _array_to_bytes(PURE_TONE_DICT[400])
+        _save_raw(filename, data)
+        self.assertTrue(filecmp.cmp(filename, exp_filename, shallow=False))
+        tmpdir.cleanup()