Changeset cd46892 for python/tests/utils.py

Timestamp:

Dec 23, 2018, 5:48:12 AM (6 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/timestretch, fix/ffmpeg5, master

Children:

51b5f9c

Parents:

966c650

Message:

[tests] add parse_file_samplerate to fetch samplerate from path

File:

• python/tests/utils.py (modified) (2 diffs)

python/tests/utils.py

@@ -2 +2 @@
 
 import os
+import re
 import glob
 import numpy as np
@@ -78 +79 @@
                     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