From b58f0bde3cb4f3ee9e01b9bb7b0d7082806ae2c6 Mon Sep 17 00:00:00 2001
From: Miguel Bazdresch <miguelb@2pif.info>
Date: Wed, 20 Sep 2023 22:08:43 -0400
Subject: [PATCH] Add tests for quinn frequency estimator.

---
 test/estimation.jl | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/test/estimation.jl b/test/estimation.jl
index 1914274e9..43395a737 100644
--- a/test/estimation.jl
+++ b/test/estimation.jl
@@ -41,15 +41,30 @@ end
     t = range(0, 5, step = 1/fs)
     fr = 28.3
     sr = cos.(2π*fr*t .+ π/4.2)
-    (f_est_real, converged, maxiter) = quinn(sr, fs)
-    @test converged
+    ### real input
+    (f_est_real, maxiter) = quinn(sr, 50, fs)
     @test maxiter == false
     @test isapprox(f_est_real, fr, atol = 1e-3)
-    # complex input
+    # use default initial guess
+    (f_est_real, maxiter) = quinn(sr, fs) # assumes f0 = 0.0
+    @test maxiter == false
+    @test isapprox(f_est_real, fr, atol = 1e-3)
+    # use default fs
+    (f_est_real, maxiter) = quinn(sr) # assumes fs = 1.0, f0 = 0.0
+    @test maxiter == false
+    @test isapprox(f_est_real, fr/fs, atol = 1e-3)
+    ### complex input
     fc = -40.3
     sc = cis.(2π*fc*t .+ π/1.4)
-    (f_est_real, converged, maxiter) = quinn(sc, fs)
-    @test converged
+    (f_est_real, maxiter) = quinn(sc, -20, fs)
     @test maxiter == false
     @test isapprox(f_est_real, fc, atol = 1e-3)
+    # use default initial guess
+    (f_est_real, maxiter) = quinn(sc, fs) # assumes f0 = 0.0
+    @test maxiter == false
+    @test isapprox(f_est_real, fc, atol = 1e-3)
+    # use default fs
+    (f_est_real, maxiter) = quinn(sc) # assumes fs = 1.0, f0 = 0.0
+    @test maxiter == false
+    @test isapprox(f_est_real, fc/fs, atol = 1e-3)
 end