4e4ac71d073b5669ea9391b28fff11e2cd7da17f
[aubio.git] / 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 def array_from_text_file(filename, dtype = 'float'):
9     filename = os.path.join(os.path.dirname(__file__), filename)
10     with open(filename) as f:
11         lines = f.readlines()
12     return np.array([line.split() for line in lines],
13             dtype = dtype)
14
15 def list_all_sounds(rel_dir):
16     datadir = os.path.join(os.path.dirname(__file__), rel_dir)
17     return glob.glob(os.path.join(datadir,'*.*'))
18
19 def get_default_test_sound(TestCase, rel_dir = 'sounds'):
20     all_sounds = list_all_sounds(rel_dir)
21     if len(all_sounds) == 0:
22         TestCase.skipTest("please add some sounds in \'python/tests/sounds\'")
23     else:
24         return all_sounds[0]
25
26 def get_tmp_sink_path():
27     fd, path = mkstemp()
28     os.close(fd)
29     return path
30
31 def del_tmp_sink_path(path):
32     os.unlink(path)
33
34 def array_from_yaml_file(filename):
35     import yaml
36     f = open(filename)
37     yaml_data = yaml.safe_load(f)
38     f.close()
39     return yaml_data
40
41 def count_samples_in_file(file_path):
42     from aubio import source
43     hopsize = 256
44     s = source(file_path, 0, hopsize)
45     total_frames = 0
46     while True:
47         _, read = s()
48         total_frames += read
49         if read < hopsize: break
50     return total_frames
51
52 def count_samples_in_directory(samples_dir):
53     total_frames = 0
54     for f in os.walk(samples_dir):
55         if len(f[2]):
56             for each in f[2]:
57                 file_path = os.path.join(f[0], each)
58                 if file_path:
59                     total_frames += count_samples_in_file(file_path)
60     return total_frames
61
62 def count_files_in_directory(samples_dir):
63     total_files = 0
64     for f in os.walk(samples_dir):
65         if len(f[2]):
66             for each in f[2]:
67                 file_path = os.path.join(f[0], each)
68                 if file_path:
69                     total_files += 1
70     return total_files