[peak_detection] Add C++ implementation of SndObj peak detection.

1 files changed, 19 insertions, 4 deletions

diff --git a/tests/test_peak_detection.py b/tests/test_peak_detection.py
index f339818..cd124f0 100644
--- a/tests/test_peak_detection.py
+++ b/tests/test_peak_detection.py
@@ -5,6 +5,7 @@ import simpl.peak_detection as peak_detection
 
 PeakDetection = peak_detection.PeakDetection
 SMSPeakDetection = peak_detection.SMSPeakDetection
+SndObjPeakDetection = peak_detection.SMSPeakDetection
 
 float_precision = 5
 frame_size = 512
@@ -59,14 +60,11 @@ class TestSMSPeakDetection(object):
         assert len(pd.frames[0].peaks)
 
     def test_size_next_read(self):
-        """
-        Make sure SMSPeakDetection is calculating the correct value for the
-        size of the next frame.
-        """
         audio, sampling_rate = simpl.read_wav(audio_path)
 
         pd = SMSPeakDetection()
         pd.hop_size = hop_size
+        pd.static_frame_size = False
         pd.max_peaks = max_peaks
         current_frame = 0
         sample_offset = 0
@@ -102,3 +100,20 @@ class TestSMSPeakDetection(object):
             assert frame.num_peaks <= max_peaks, frame.num_peaks
             max_amp = max([p.amplitude for p in frame.peaks])
             assert max_amp
+
+
+class TestSndObjPeakDetection(object):
+    def test_peak_detection(self):
+        audio, sampling_rate = simpl.read_wav(audio_path)
+
+        pd = SndObjPeakDetection()
+        pd.max_peaks = max_peaks
+        pd.hop_size = hop_size
+        frames = pd.find_peaks(audio[0:num_samples])
+
+        assert len(frames) == num_samples / hop_size
+
+        for frame in frames:
+            assert frame.num_peaks <= max_peaks, frame.num_peaks
+            max_amp = max([p.amplitude for p in frame.peaks])
+            assert max_amp