python/tests/utils.py

   1 #! /usr/bin/env python
   2
   3 import os
   4 import glob
   5 import numpy as np
   6 from tempfile import mkstemp
   7
   8 DEFAULT_SOUND = '22050Hz_5s_brownnoise.wav'
   9
  10 def array_from_text_file(filename, dtype = 'float'):
  11     realpathname = os.path.join(os.path.dirname(__file__), filename)
  12     return np.loadtxt(realpathname, dtype = dtype)
  13
  14 def list_all_sounds(rel_dir):
  15     datadir = os.path.join(os.path.dirname(__file__), rel_dir)
  16     return glob.glob(os.path.join(datadir,'*.*'))
  17
  18 def get_default_test_sound(TestCase, rel_dir = 'sounds'):
  19     all_sounds = list_all_sounds(rel_dir)
  20     if len(all_sounds) == 0:
  21         TestCase.skipTest("please add some sounds in \'python/tests/sounds\'")
  22     else:
  23         default_sound = all_sounds[0]
  24         if DEFAULT_SOUND in map(os.path.basename, all_sounds):
  25             while os.path.basename(default_sound) != DEFAULT_SOUND:
  26                 default_sound = all_sounds.pop(0)
  27         return default_sound
  28
  29 def get_tmp_sink_path():
  30     fd, path = mkstemp()
  31     os.close(fd)
  32     return path
  33
  34 def del_tmp_sink_path(path):
  35     try:
  36         os.unlink(path)
  37     except WindowsError as e:
  38         # removing the temporary directory sometimes fails on windows
  39         import warnings
  40         errmsg = "failed deleting temporary file {:s} ({:s})"
  41         warnings.warn(UserWarning(errmsg.format(path, repr(e))))
  42
  43 def array_from_yaml_file(filename):
  44     import yaml
  45     f = open(filename)
  46     yaml_data = yaml.safe_load(f)
  47     f.close()
  48     return yaml_data
  49
  50 def count_samples_in_file(file_path):
  51     from aubio import source
  52     hopsize = 256
  53     s = source(file_path, 0, hopsize)
  54     total_frames = 0
  55     while True:
  56         _, read = s()
  57         total_frames += read
  58         if read < hopsize: break
  59     return total_frames
  60
  61 def count_samples_in_directory(samples_dir):
  62     total_frames = 0
  63     for f in os.walk(samples_dir):
  64         if len(f[2]):
  65             for each in f[2]:
  66                 file_path = os.path.join(f[0], each)
  67                 if file_path:
  68                     total_frames += count_samples_in_file(file_path)
  69     return total_frames
  70
  71 def count_files_in_directory(samples_dir):
  72     total_files = 0
  73     for f in os.walk(samples_dir):
  74         if len(f[2]):
  75             for each in f[2]:
  76                 file_path = os.path.join(f[0], each)
  77                 if file_path:
  78                     total_files += 1
  79     return total_files