treatment of empty spike trains in spike sync

1 files changed, 39 insertions, 0 deletions

diff --git a/test/test_empty.py b/test/test_empty.py
index 42c3716..b31d8a4 100644
--- a/test/test_empty.py
+++ b/test/test_empty.py
@@ -102,7 +102,46 @@ def test_spike_empty():
     assert_array_almost_equal(prof.y2, expected_y2, decimal=15)
 
 
+def test_spike_sync_empty():
+    st1 = SpikeTrain([], edges=(0.0, 1.0))
+    st2 = SpikeTrain([], edges=(0.0, 1.0))
+    d = spk.spike_sync(st1, st2)
+    assert_equal(d, 0.0)
+    prof = spk.spike_sync_profile(st1, st2)
+    assert_equal(d, prof.avrg())
+    assert_array_equal(prof.x, [0.0, 1.0])
+    assert_array_equal(prof.y, [0.0, 0.0])
+
+    st1 = SpikeTrain([], edges=(0.0, 1.0))
+    st2 = SpikeTrain([0.4, ], edges=(0.0, 1.0))
+    d = spk.spike_sync(st1, st2)
+    assert_equal(d, 0.0)
+    prof = spk.spike_sync_profile(st1, st2)
+    assert_equal(d, prof.avrg())
+    assert_array_equal(prof.x, [0.0, 0.4, 1.0])
+    assert_array_equal(prof.y, [0.0, 0.0, 0.0])
+
+    st1 = SpikeTrain([0.6, ], edges=(0.0, 1.0))
+    st2 = SpikeTrain([0.4, ], edges=(0.0, 1.0))
+    d = spk.spike_sync(st1, st2)
+    assert_almost_equal(d, 1.0, decimal=15)
+    prof = spk.spike_sync_profile(st1, st2)
+    assert_equal(d, prof.avrg())
+    assert_array_almost_equal(prof.x, [0.0, 0.4, 0.6, 1.0], decimal=15)
+    assert_array_almost_equal(prof.y, [1.0, 1.0, 1.0, 1.0], decimal=15)
+
+    st1 = SpikeTrain([0.2, ], edges=(0.0, 1.0))
+    st2 = SpikeTrain([0.8, ], edges=(0.0, 1.0))
+    d = spk.spike_sync(st1, st2)
+    assert_almost_equal(d, 0.0, decimal=15)
+    prof = spk.spike_sync_profile(st1, st2)
+    assert_equal(d, prof.avrg())
+    assert_array_almost_equal(prof.x, [0.0, 0.2, 0.8, 1.0], decimal=15)
+    assert_array_almost_equal(prof.y, [0.0, 0.0, 0.0, 0.0], decimal=15)
+
+
 if __name__ == "__main__":
     test_get_non_empty()
     test_isi_empty()
     test_spike_empty()
+    test_spike_sync_empty()