1 | #! /usr/bin/env python |
---|
2 | |
---|
3 | from numpy.testing import TestCase, assert_equal, assert_almost_equal |
---|
4 | from aubio import notes, source |
---|
5 | import numpy as np |
---|
6 | from utils import list_all_sounds |
---|
7 | |
---|
8 | list_of_sounds = list_all_sounds('sounds') |
---|
9 | |
---|
10 | AUBIO_DEFAULT_NOTES_SILENCE = -70. |
---|
11 | AUBIO_DEFAULT_NOTES_RELEASE_DROP = 10. |
---|
12 | AUBIO_DEFAULT_NOTES_MINIOI_MS = 30. |
---|
13 | |
---|
14 | class aubio_notes_default(TestCase): |
---|
15 | |
---|
16 | def test_members(self): |
---|
17 | o = notes() |
---|
18 | assert_equal ([o.buf_size, o.hop_size, o.method, o.samplerate], |
---|
19 | [1024,512,'default',44100]) |
---|
20 | |
---|
21 | |
---|
22 | class aubio_notes_params(TestCase): |
---|
23 | |
---|
24 | samplerate = 44100 |
---|
25 | |
---|
26 | def setUp(self): |
---|
27 | self.o = notes(samplerate = self.samplerate) |
---|
28 | |
---|
29 | def test_get_minioi_ms(self): |
---|
30 | assert_equal (self.o.get_minioi_ms(), AUBIO_DEFAULT_NOTES_MINIOI_MS) |
---|
31 | |
---|
32 | def test_set_minioi_ms(self): |
---|
33 | val = 40. |
---|
34 | self.o.set_minioi_ms(val) |
---|
35 | assert_almost_equal (self.o.get_minioi_ms(), val) |
---|
36 | |
---|
37 | def test_get_silence(self): |
---|
38 | assert_equal (self.o.get_silence(), AUBIO_DEFAULT_NOTES_SILENCE) |
---|
39 | |
---|
40 | def test_set_silence(self): |
---|
41 | val = -50 |
---|
42 | self.o.set_silence(val) |
---|
43 | assert_equal (self.o.get_silence(), val) |
---|
44 | |
---|
45 | def test_get_release_drop(self): |
---|
46 | assert_equal (self.o.get_release_drop(), AUBIO_DEFAULT_NOTES_RELEASE_DROP) |
---|
47 | |
---|
48 | def test_set_release_drop(self): |
---|
49 | val = 50 |
---|
50 | self.o.set_release_drop(val) |
---|
51 | assert_equal (self.o.get_release_drop(), val) |
---|
52 | |
---|
53 | def test_set_release_drop_wrong(self): |
---|
54 | val = -10 |
---|
55 | with self.assertRaises(ValueError): |
---|
56 | self.o.set_release_drop(val) |
---|
57 | |
---|
58 | class aubio_notes_sinewave(TestCase): |
---|
59 | |
---|
60 | def analyze_file(self, filepath, samplerate=0): |
---|
61 | win_s = 512 # fft size |
---|
62 | hop_s = 256 # hop size |
---|
63 | |
---|
64 | s = source(filepath, samplerate, hop_s) |
---|
65 | samplerate = s.samplerate |
---|
66 | |
---|
67 | tolerance = 0.8 |
---|
68 | |
---|
69 | notes_o = notes("default", win_s, hop_s, samplerate) |
---|
70 | total_frames = 0 |
---|
71 | |
---|
72 | results = [] |
---|
73 | while True: |
---|
74 | samples, read = s() |
---|
75 | new_note = notes_o(samples) |
---|
76 | if (new_note[0] != 0): |
---|
77 | note_str = ' '.join(["%.2f" % i for i in new_note]) |
---|
78 | results.append( [total_frames, np.copy(new_note)] ) |
---|
79 | total_frames += read |
---|
80 | if read < hop_s: break |
---|
81 | return results |
---|
82 | |
---|
83 | def test_sinewave(self): |
---|
84 | for filepath in list_of_sounds: |
---|
85 | if '44100Hz_44100f_sine441.wav' in filepath: |
---|
86 | results = self.analyze_file(filepath) |
---|
87 | assert_equal (len(results), 1) |
---|
88 | assert_equal (len(results[0]), 2) |
---|
89 | assert_equal (results[0][0], 1280) |
---|
90 | assert_equal (results[0][1], [69, 123, -1]) |
---|
91 | |
---|
92 | if __name__ == '__main__': |
---|
93 | from unittest import main |
---|
94 | main() |
---|