diff options
| author | John Glover <j@johnglover.net> | 2012-07-02 20:42:47 +0100 |
|---|---|---|
| committer | John Glover <j@johnglover.net> | 2012-07-02 20:42:47 +0100 |
| commit | 2675f0ef49584ceb03283430716e053fe0498a3a (patch) | |
| tree | 05d4088fe3e7550a9bc8041f4f7839ad21ccd92a /tests/test_partial_tracking.py | |
| parent | 9312f0a18a208c4dde6c400b1e08b678648be588 (diff) | |
| download | simpl-2675f0ef49584ceb03283430716e053fe0498a3a.tar.gz simpl-2675f0ef49584ceb03283430716e053fe0498a3a.tar.bz2 simpl-2675f0ef49584ceb03283430716e053fe0498a3a.zip | |
[partial_tracking] Add C++ implemention of SMS partial tracking.
Update libsms test data.
Diffstat (limited to 'tests/test_partial_tracking.py')
| -rw-r--r-- | tests/test_partial_tracking.py | 36 |
1 files changed, 20 insertions, 16 deletions
diff --git a/tests/test_partial_tracking.py b/tests/test_partial_tracking.py index fb87da9..32f2653 100644 --- a/tests/test_partial_tracking.py +++ b/tests/test_partial_tracking.py @@ -1,6 +1,5 @@ import os import json -import numpy as np from nose.tools import assert_almost_equals import simpl import simpl.peak_detection as peak_detection @@ -11,7 +10,7 @@ SMSPeakDetection = peak_detection.SMSPeakDetection PartialTracking = partial_tracking.PartialTracking SMSPartialTracking = partial_tracking.SMSPartialTracking -float_precision = 5 +float_precision = 2 frame_size = 512 hop_size = 512 max_peaks = 10 @@ -77,7 +76,7 @@ class TestSMSPartialTracking(object): (len(frames), len(self.audio) / hop_size) assert len(frames) == len(self.audio) / hop_size - assert len(frames[0].partials) == 0 + assert frames[0].num_partials == max_partials assert frames[0].max_partials == max_partials def test_partial_tracking(self): @@ -89,16 +88,21 @@ class TestSMSPartialTracking(object): pt.max_partials = max_partials frames = pt.find_partials(peaks) - # make sure each partial is the same - # for i in range(len(sms_frames)): - # assert len(sms_frames[i].partials) == len(simpl_frames[i].partials) - # for p in range(len(sms_frames[i].partials)): - # assert_almost_equals(sms_frames[i].partials[p].amplitude, - # simpl_frames[i].partials[p].amplitude, - # float_precision) - # assert_almost_equals(sms_frames[i].partials[p].frequency, - # simpl_frames[i].partials[p].frequency, - # float_precision) - # assert_almost_equals(sms_frames[i].partials[p].phase, - # simpl_frames[i].partials[p].phase, - # float_precision) + sms_frames = self.test_data['partial_tracking'] + sms_frames = sms_frames[0:len(sms_frames) - 3] + + assert len(sms_frames) == len(frames) + + for i in range(len(frames)): + assert len(frames[i].partials) == len(sms_frames[i]['partials']) + + for p in range(len(frames[i].partials)): + assert_almost_equals(frames[i].partials[p].amplitude, + sms_frames[i]['partials'][p]['amplitude'], + float_precision) + assert_almost_equals(frames[i].partials[p].frequency, + sms_frames[i]['partials'][p]['frequency'], + float_precision) + assert_almost_equals(frames[i].partials[p].phase, + sms_frames[i]['partials'][p]['phase'], + float_precision) |