new fft

tests/test_ring_buffer.py

@@ -56,6 +56,20 @@ class RingBufferTests(unittest.TestCase):
         self.assertFalse(np.allclose(ring.last_fft_db, fft_before))
         self.assertFalse(np.allclose(ring.distance_axis, axis_before))
 
+    def test_ring_buffer_can_switch_to_positive_only_fft_mode(self):
+        ring = RingBuffer(max_sweeps=2)
+        sweep = np.linspace(0.0, 1.0, 64, dtype=np.float32)
+        freqs = np.linspace(3.3, 14.3, 64, dtype=np.float64)
+        ring.push(sweep, freqs)
+
+        changed = ring.set_fft_mode("positive_only")
+
+        self.assertTrue(changed)
+        self.assertEqual(ring.fft_mode, "positive_only")
+        self.assertIsNotNone(ring.last_fft_db)
+        self.assertEqual(ring.last_fft_db.shape, (ring.fft_bins,))
+        self.assertIsNotNone(ring.distance_axis)
+
 
 if __name__ == "__main__":
     unittest.main()