Refactor frequency assertion for improved performance

src/jimgw/core/single_event/data.py

@@ -315,7 +315,7 @@ def from_fd(
         delta_t = 1 / (2 * fnyq)
         data_td_full = jnp.fft.irfft(data_fd_full) / delta_t
         # check frequencies
-        assert jnp.allclose(
+        assert jnp.array_equal(
             f, jnp.fft.rfftfreq(len(data_td_full), delta_t)
         ), "Generated frequencies do not match the input frequencies"
         # create a Data object
@@ -325,9 +325,9 @@ def from_fd(
         # This ensures the newly constructed Data in FD fully
         # represents the input FD data.
         d_new, f_new = data.frequency_slice(frequencies[0], frequencies[-1])
-        assert all(jnp.equal(d_new, fd)), "Data do not match after slicing"
-        assert all(
-            jnp.equal(f_new, frequencies)
+        assert jnp.array_equal(d_new, fd), "Data do not match after slicing"
+        assert jnp.array_equal(
+            f_new, frequencies
         ), "Frequencies do not match after slicing"
         return data
 

src/jimgw/core/single_event/detector.py

@@ -142,8 +142,8 @@ def set_frequency_bounds(
         data, freqs_1 = self.data.frequency_slice(*self.frequency_bounds)
         psd, freqs_2 = self.psd.frequency_slice(*self.frequency_bounds)
 
-        assert all(
-            freqs_1 == freqs_2
+        assert jnp.array_equal(
+            freqs_1, freqs_2
         ), f"The {self.name} data and PSD must have same frequencies"
 
         self._sliced_frequencies = freqs_1