Fix test of lomb-scargle periodogram with wrong comparison

StingraySoftware · Oct 7, 2023 · cf4f9f6 · cf4f9f6
1 parent fdad7d7
commit cf4f9f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stingray/tests/test_lombscargle.py b/stingray/tests/test_lombscargle.py
@@ -59,7 +59,7 @@ def test_eventlist(self):
         ev_lscs = LombScargleCrossspectrum(ev1, ev2, dt=1)
         lc_lscs = LombScargleCrossspectrum(lc1, lc2, dt=1)
 
-        assert np.argmax(lc_lscs) == np.argmax(ev_lscs)
+        assert np.argmax(lc_lscs.power) == np.argmax(ev_lscs.power)
         assert np.all(ev_lscs.freq == lc_lscs.freq)
         assert np.all(ev_lscs.power == lc_lscs.power)
         assert ev_lscs.freq[np.argmax(ev_lscs.power)] == lc_lscs.freq[np.argmax(lc_lscs.power)] != 0