use allclose

dylansdaniels · Jul 9, 2024 · 71f03b2 · 71f03b2
1 parent 0bc2c1f
commit 71f03b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hnn_core/tests/test_drives.py b/hnn_core/tests/test_drives.py
@@ -113,7 +113,7 @@ def test_drive_seeds(setup_net):
     trial2_spikes = np.array(sorted(
         net.external_drives['prox']['events'][1]))
     # No two spikes should be perfectly identical across seeds
-    assert ~np.any(trial1_spikes == trial2_spikes)
+    assert ~np.any(np.allclose(trial1_spikes, trial2_spikes))
 
 
 def test_add_drives():