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     filename = os.path.join(os.path.dirname(__file__), filename)
  12     with open(filename) as f:
  13         lines = f.readlines()
  14     return np.array([line.split() for line in lines],
  15             dtype = dtype)
  16
  17 def list_all_sounds(rel_dir):
  18     datadir = os.path.join(os.path.dirname(__file__), rel_dir)
  19     return glob.glob(os.path.join(datadir,'*.*'))
  20
  21 def get_default_test_sound(TestCase, rel_dir = 'sounds'):
  22     all_sounds = list_all_sounds(rel_dir)
  23     if len(all_sounds) == 0:
  24         TestCase.skipTest("please add some sounds in \'python/tests/sounds\'")
  25     else:
  26         default_sound = all_sounds[0]
  27         if DEFAULT_SOUND in map(os.path.basename, all_sounds):
  28             while os.path.basename(default_sound) != DEFAULT_SOUND:
  29                 default_sound = all_sounds.pop(0)
  30         return default_sound
  31
  32 def get_tmp_sink_path():
  33     fd, path = mkstemp()
  34     os.close(fd)
  35     return path
  36
  37 def del_tmp_sink_path(path):
  38     try:
  39         os.unlink(path)
  40     except WindowsError as e:
  41         print("deleting {:s} failed ({:s}), reopening".format(path, repr(e)))
  42         with open(path, 'wb') as f:
  43             f.close()
  44         try:
  45             os.unlink(path)
  46         except WindowsError as f:
  47             print("deleting {:s} failed ({:s}), aborting".format(path, repr(e)))
  48
  49 def array_from_yaml_file(filename):
  50     import yaml
  51     f = open(filename)
  52     yaml_data = yaml.safe_load(f)
  53     f.close()
  54     return yaml_data
  55
  56 def count_samples_in_file(file_path):
  57     from aubio import source
  58     hopsize = 256
  59     s = source(file_path, 0, hopsize)
  60     total_frames = 0
  61     while True:
  62         _, read = s()
  63         total_frames += read
  64         if read < hopsize: break
  65     return total_frames
  66
  67 def count_samples_in_directory(samples_dir):
  68     total_frames = 0
  69     for f in os.walk(samples_dir):
  70         if len(f[2]):
  71             for each in f[2]:
  72                 file_path = os.path.join(f[0], each)
  73                 if file_path:
  74                     total_frames += count_samples_in_file(file_path)
  75     return total_frames
  76
  77 def count_files_in_directory(samples_dir):
  78     total_files = 0
  79     for f in os.walk(samples_dir):
  80         if len(f[2]):
  81             for each in f[2]:
  82                 file_path = os.path.join(f[0], each)
  83                 if file_path:
  84                     total_files += 1
  85     return total_files