[py] add helper to check if we are on a 32bit system

[aubio.git] / python / tests / utils.py

diff --git a/python/tests/utils.py b/python/tests/utils.py
index fda8cb7..7606404 100644 (file)
--- a/python/tests/utils.py
+++ b/python/tests/utils.py
@@ -1,20 +1,97 @@
 #! /usr/bin/env python
 
+import os
+import re
+import glob
+import struct
+import numpy as np
+from tempfile import mkstemp
+
+DEFAULT_SOUND = '22050Hz_5s_brownnoise.wav'
+
+def is32bit():
+    return struct.calcsize("P") * 8 == 32
+
 def array_from_text_file(filename, dtype = 'float'):
-    import os.path
-    from numpy import array
-    filename = os.path.join(os.path.dirname(__file__), filename)
-    return array([line.split() for line in open(filename).readlines()],
-        dtype = dtype)
+    realpathname = os.path.join(os.path.dirname(__file__), filename)
+    return np.loadtxt(realpathname, dtype = dtype)
 
 def list_all_sounds(rel_dir):
-    import os.path, glob
     datadir = os.path.join(os.path.dirname(__file__), rel_dir)
     return glob.glob(os.path.join(datadir,'*.*'))
 
+def get_default_test_sound(TestCase, rel_dir = 'sounds'):
+    all_sounds = list_all_sounds(rel_dir)
+    if len(all_sounds) == 0:
+        TestCase.skipTest("please add some sounds in \'python/tests/sounds\'")
+    else:
+        default_sound = all_sounds[0]
+        if DEFAULT_SOUND in map(os.path.basename, all_sounds):
+            while os.path.basename(default_sound) != DEFAULT_SOUND:
+                default_sound = all_sounds.pop(0)
+        return default_sound
+
+def get_tmp_sink_path():
+    fd, path = mkstemp()
+    os.close(fd)
+    return path
+
+def del_tmp_sink_path(path):
+    try:
+        os.unlink(path)
+    except WindowsError as e:
+        # removing the temporary directory sometimes fails on windows
+        import warnings
+        errmsg = "failed deleting temporary file {:s} ({:s})"
+        warnings.warn(UserWarning(errmsg.format(path, repr(e))))
+
 def array_from_yaml_file(filename):
     import yaml
     f = open(filename)
     yaml_data = yaml.safe_load(f)
     f.close()
     return yaml_data
+
+def count_samples_in_file(file_path):
+    from aubio import source
+    hopsize = 256
+    s = source(file_path, 0, hopsize)
+    total_frames = 0
+    while True:
+        _, read = s()
+        total_frames += read
+        if read < hopsize: break
+    return total_frames
+
+def count_samples_in_directory(samples_dir):
+    total_frames = 0
+    for f in os.walk(samples_dir):
+        if len(f[2]):
+            for each in f[2]:
+                file_path = os.path.join(f[0], each)
+                if file_path:
+                    total_frames += count_samples_in_file(file_path)
+    return total_frames
+
+def count_files_in_directory(samples_dir):
+    total_files = 0
+    for f in os.walk(samples_dir):
+        if len(f[2]):
+            for each in f[2]:
+                file_path = os.path.join(f[0], each)
+                if file_path:
+                    total_files += 1
+    return total_files
+
+def parse_file_samplerate(soundfile):
+    samplerate = None
+    # parse samplerate
+    re_sr = re.compile(r'/([0-9]{4,})Hz_.*')
+    match_samplerate = re_sr.findall(soundfile)
+    if match_samplerate:
+        samplerate = int(match_samplerate[0])
+    else:
+        import warnings
+        warnings.warn(UserWarning("could not parse samplerate for {:s}"
+            .format(soundfile)))
+    return samplerate