source: python/tests/utils.py @ 45f48576

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since 45f48576 was ab35262, checked in by Paul Brossier <piem@piem.org>, 11 years ago

python/tests/test_sink.py: add more tests, clean up created files

  • Property mode set to 100644
File size: 1.9 KB
Line 
1#! /usr/bin/env python
2
3def array_from_text_file(filename, dtype = 'float'):
4    import os.path
5    from numpy import array
6    filename = os.path.join(os.path.dirname(__file__), filename)
7    return array([line.split() for line in open(filename).readlines()],
8        dtype = dtype)
9
10def list_all_sounds(rel_dir):
11    import os.path, glob
12    datadir = os.path.join(os.path.dirname(__file__), rel_dir)
13    return glob.glob(os.path.join(datadir,'*.*'))
14
15def get_default_test_sound(TestCase, rel_dir = 'sounds'):
16    all_sounds = list_all_sounds(rel_dir)
17    if len(all_sounds) == 0:
18        TestCase.skipTest("please add some sounds in \'python/tests/sounds\'")
19    else:
20        return all_sounds[0]
21
22def get_tmp_sink_path():
23    from tempfile import mkstemp
24    import os
25    fd, path = mkstemp()
26    os.close(fd)
27    return path
28
29def del_tmp_sink_path(path):
30    import os
31    os.unlink(path)
32
33def array_from_yaml_file(filename):
34    import yaml
35    f = open(filename)
36    yaml_data = yaml.safe_load(f)
37    f.close()
38    return yaml_data
39
40def count_samples_in_file(file_path):
41    from aubio import source
42    hopsize = 256
43    s = source(file_path, 0, hopsize)
44    total_frames = 0
45    while True:
46        samples, read = s()
47        total_frames += read
48        if read < hopsize: break
49    return total_frames
50
51def count_samples_in_directory(samples_dir):
52    import os
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
62def count_files_in_directory(samples_dir):
63    import os
64    total_files = 0
65    for f in os.walk(samples_dir):
66        if len(f[2]):
67            for each in f[2]:
68                file_path = os.path.join(f[0], each)
69                if file_path:
70                    total_files += 1
71    return total_files
Note: See TracBrowser for help on using the repository browser.