[partial_tracking] Call Cython PartialTracking.update_partials

instead of C++ method so that Python classes
can inherit it.

Save frame partials as a Python list instead of
getting a list of peaks from the C++ object.

Update partial tracking tests.

1 files changed, 2 insertions, 9 deletions

diff --git a/tests/test_partial_tracking.py b/tests/test_partial_tracking.py
index 32f2653..cab6b2b 100644
--- a/tests/test_partial_tracking.py
+++ b/tests/test_partial_tracking.py
@@ -46,10 +46,6 @@ class TestPartialTracking(object):
         pt = PartialTracking()
         frames = pt.find_partials(frames)
 
-        print 'frames: %d (expected: %d)' %\
-            (len(frames), len(self.audio) / hop_size)
-        assert len(frames) == len(self.audio) / hop_size
-
         assert len(frames[0].partials) == 0
         assert frames[0].max_partials == 100
 
@@ -64,6 +60,7 @@ class TestSMSPartialTracking(object):
     def test_basic(self):
         pd = SMSPeakDetection()
         pd.hop_size = hop_size
+        pd.frame_size = hop_size
         pd.max_peaks = max_peaks
         pd.static_frame_size = True
         frames = pd.find_peaks(self.audio)
@@ -72,11 +69,7 @@ class TestSMSPartialTracking(object):
         pt.max_partials = max_partials
         frames = pt.find_partials(frames)
 
-        print 'frames: %d (expected: %d)' %\
-            (len(frames), len(self.audio) / hop_size)
-        assert len(frames) == len(self.audio) / hop_size
-
-        assert frames[0].num_partials == max_partials
+        assert len(frames[0].partials) == max_partials
         assert frames[0].max_partials == max_partials
 
     def test_partial_tracking(self):