ref: 40792b2ef7f967e07626b22f6f120397a5f76317
dir: /python/tests/test_notes.py/
#! /usr/bin/env python
from unittest import main
from numpy.testing import TestCase, assert_equal, assert_almost_equal
from aubio import notes
AUBIO_DEFAULT_NOTES_SILENCE = -70.
AUBIO_DEFAULT_NOTES_MINIOI_MS = 30.
class aubio_notes_default(TestCase):
def test_members(self):
o = notes()
assert_equal ([o.buf_size, o.hop_size, o.method, o.samplerate],
[1024,512,'default',44100])
class aubio_notes_params(TestCase):
samplerate = 44100
def setUp(self):
self.o = notes(samplerate = self.samplerate)
def test_get_minioi_ms(self):
assert_equal (self.o.get_minioi_ms(), AUBIO_DEFAULT_NOTES_MINIOI_MS)
def test_set_minioi_ms(self):
val = 40.
self.o.set_minioi_ms(val)
assert_almost_equal (self.o.get_minioi_ms(), val)
def test_get_silence(self):
assert_equal (self.o.get_silence(), AUBIO_DEFAULT_NOTES_SILENCE)
def test_set_silence(self):
val = -50
self.o.set_silence(val)
assert_equal (self.o.get_silence(), val)
from .utils import list_all_sounds
list_of_sounds = list_all_sounds('sounds')
class aubio_notes_sinewave(TestCase):
def analyze_file(self, filepath, samplerate=0):
from aubio import source
import numpy as np
win_s = 512 # fft size
hop_s = 256 # hop size
s = source(filepath, samplerate, hop_s)
samplerate = s.samplerate
tolerance = 0.8
notes_o = notes("default", win_s, hop_s, samplerate)
total_frames = 0
results = []
while True:
samples, read = s()
new_note = notes_o(samples)
if (new_note[0] != 0):
note_str = ' '.join(["%.2f" % i for i in new_note])
results.append( [total_frames, np.copy(new_note)] )
total_frames += read
if read < hop_s: break
return results
def test_sinewave(self):
for filepath in list_of_sounds:
if '44100Hz_44100f_sine441.wav' in filepath:
results = self.analyze_file(filepath)
assert_equal (len(results), 1)
assert_equal (len(results[0]), 2)
assert_equal (results[0][0], 1280)
assert_equal (results[0][1], [69, 123, -1])
if __name__ == '__main__':
main()