comparison tests/test_io.py @ 321:903b5e1e8af9

Clean up tests

 import filecmp
 from unittest import TestCase
 from unittest.mock import patch, Mock
 from genty import genty, genty_dataset
 from test_util import _sample_generator, _generate_pure_tone, PURE_TONE_DICT
+from auditok.signal import FORMAT
 from auditok.io import (
-    DATA_FORMAT,
     AudioIOError,
     AudioParameterError,
     BufferAudioSource,
     RawAudioSource,
     WaveAudioSource,

         self.assertIsInstance(audio_source, expected_class)
         self.assertEqual(audio_source.sampling_rate, 16000)
         self.assertEqual(audio_source.sample_width, 2)
         self.assertEqual(audio_source.channels, len(frequencies))
         mono_channels = [PURE_TONE_DICT[freq] for freq in frequencies]
-        fmt = DATA_FORMAT[audio_source.sample_width]
+        fmt = FORMAT[audio_source.sample_width]
         expected = array(fmt, _sample_generator(*mono_channels)).tobytes()
         self.assertEqual(data, expected)
 
     @genty_dataset(
         missing_sampling_rate=("sr",),

         self.assertIsInstance(audio_source, expected_class)
         self.assertEqual(audio_source.sampling_rate, 16000)
         self.assertEqual(audio_source.sample_width, 2)
         self.assertEqual(audio_source.channels, len(frequencies))
         mono_channels = [PURE_TONE_DICT[freq] for freq in frequencies]
-        fmt = DATA_FORMAT[audio_source.sample_width]
+        fmt = FORMAT[audio_source.sample_width]
         expected = array(fmt, _sample_generator(*mono_channels)).tobytes()
         self.assertEqual(data, expected)
 
     @patch("auditok.io._WITH_PYDUB", True)
     @patch("auditok.io.BufferAudioSource")

         three_channel=("3channel_400-800-1600Hz.raw", (400, 800, 1600)),
     )
     def test_save_raw(self, filename, frequencies):
         filename = "tests/data/test_16KHZ_{}".format(filename)
         sample_width = 2
-        fmt = DATA_FORMAT[sample_width]
+        fmt = FORMAT[sample_width]
         mono_channels = [PURE_TONE_DICT[freq] for freq in frequencies]
         data = array(fmt, _sample_generator(*mono_channels)).tobytes()
         tmpfile = NamedTemporaryFile()
         _save_raw(data, tmpfile.name)
         self.assertTrue(filecmp.cmp(tmpfile.name, filename, shallow=False))

     def test_save_wave(self, filename, frequencies):
         filename = "tests/data/test_16KHZ_{}".format(filename)
         sampling_rate = 16000
         sample_width = 2
         channels = len(frequencies)
-        fmt = DATA_FORMAT[sample_width]
+        fmt = FORMAT[sample_width]
         mono_channels = [PURE_TONE_DICT[freq] for freq in frequencies]
         data = array(fmt, _sample_generator(*mono_channels)).tobytes()
         tmpfile = NamedTemporaryFile()
         _save_wave(data, tmpfile.name, sampling_rate, sample_width, channels)
         self.assertTrue(filecmp.cmp(tmpfile.name, filename, shallow=False))