From 1330c8d4b6995d5a9946b269c9747a11e9216ef1 Mon Sep 17 00:00:00 2001
From: Chris Cannam <cannam@all-day-breakfast.com>
Date: Wed, 15 Jun 2022 10:17:43 +0100
Subject: [PATCH] Add tests for p=1 and p=0

---
 src/test/TestSignalBits.cpp | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/test/TestSignalBits.cpp b/src/test/TestSignalBits.cpp
index 8c6e106..bd56248 100644
--- a/src/test/TestSignalBits.cpp
+++ b/src/test/TestSignalBits.cpp
@@ -227,6 +227,28 @@ BOOST_AUTO_TEST_CASE(peakpick_nearest_2_12)
     BOOST_TEST(out == expected, tt::per_element());
 }
 
+BOOST_AUTO_TEST_CASE(peakpick_nearest_1_12)
+{
+    Peak<double> pp(12);
+    vector<double> in { -0.3, -0.1, -0.2, 1.0, -0.3, -0.5,
+                        -0.5, -0.4, -0.1, -0.1, -0.2, -0.3 };
+    vector<int> out(12);
+    vector<int> expected { 1, 1, 3, 3, 3, 3, 8, 8, 8, 8, 8, 8 };
+    pp.findNearestAndNextPeaks(in.data(), 1, out.data(), nullptr);
+    BOOST_TEST(out == expected, tt::per_element());
+}
+
+BOOST_AUTO_TEST_CASE(peakpick_nearest_0_12)
+{
+    Peak<double> pp(12);
+    vector<double> in { -0.3, -0.1, -0.2, 1.0, -0.3, -0.5,
+                        -0.5, -0.4, -0.1, -0.1, -0.2, -0.3 };
+    vector<int> out(12);
+    vector<int> expected { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 };
+    pp.findNearestAndNextPeaks(in.data(), 0, out.data(), nullptr);
+    BOOST_TEST(out == expected, tt::per_element());
+}
+
 BOOST_AUTO_TEST_CASE(peakpick_next_2_1)
 {
     Peak<double> pp(1);