[loris] LorisPeakDetection updates and fixes (save bandwidth to simpl Peak objects).

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_peak_detection.cpp b/tests/test_peak_detection.cpp
index 41bc735..4c09dbd 100644
--- a/tests/test_peak_detection.cpp
+++ b/tests/test_peak_detection.cpp
@@ -31,39 +31,54 @@ protected:
     int num_samples;
 
     void test_find_peaks_in_frame_basic() {
+        pd->clear();
+        pd->frame_size(2048);
+
         Frame* f = new Frame(2048, true);
         Peaks p = pd->find_peaks_in_frame(f);
         CPPUNIT_ASSERT(p.size() == 0);
+
         delete f;
+        pd->clear();
     }
 
     void test_find_peaks_basic() {
         sample* audio = new sample[1024];
+        pd->frame_size(512);
+
         Frames frames = pd->find_peaks(1024, audio);
         CPPUNIT_ASSERT(frames.size() == 2);
         for(int i = 0; i < frames.size(); i++) {
             CPPUNIT_ASSERT(frames[i]->num_peaks() == 0);
         }
+
+        delete audio;
     }
 
     void test_find_peaks_change_hop_frame_size() {
         sample* audio = new sample[1024];
         pd->frame_size(256);
         pd->hop_size(256);
+
         Frames frames = pd->find_peaks(1024, audio);
         CPPUNIT_ASSERT(frames.size() == 4);
         for(int i = 0; i < frames.size(); i++) {
             CPPUNIT_ASSERT(frames[i]->num_peaks() == 0);
         }
+
+        delete audio;
     }
 
     void test_find_peaks_audio() {
         sample* audio = new sample[(int)sf.frames()];
         sf.read(audio, (int)sf.frames());
+
         Frames frames = pd->find_peaks(num_samples, &(audio[(int)sf.frames() / 2]));
         for(int i = 0; i < frames.size(); i++) {
             CPPUNIT_ASSERT(frames[i]->num_peaks() > 0);
         }
+
+        delete audio;
     }
 
 public: