diff tests/test_util.py @ 321:903b5e1e8af9

Clean up tests
author Amine Sehili <amine.sehili@gmail.com>
date Sat, 19 Oct 2019 15:01:38 +0100
parents 18a9f0dcdaae
children 9f17aa9a4018
line wrap: on
line diff
--- a/tests/test_util.py	Fri Oct 18 21:56:43 2019 +0100
+++ b/tests/test_util.py	Sat Oct 19 15:01:38 2019 +0100
@@ -3,7 +3,7 @@
 from array import array
 from genty import genty, genty_dataset
 from auditok.util import AudioEnergyValidator
-from auditok.io import DATA_FORMAT
+from auditok.signal import FORMAT
 
 
 def _sample_generator(*data_buffers):
@@ -29,7 +29,7 @@
     max_value = (2 ** (sample_width * 8) // 2) - 1
     if volume > max_value:
         volume = max_value
-    fmt = DATA_FORMAT[sample_width]
+    fmt = FORMAT[sample_width]
     total_samples = int(sampling_rate * duration_sec)
     step = frequency / sampling_rate
     two_pi_step = 2 * math.pi * step