diff options
author | Mario Mulansky <mario.mulansky@gmx.net> | 2018-02-10 12:48:38 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-02-10 12:48:38 -0800 |
commit | 9d7893d8f4f7a1c20dfab2512c441222714e5f7e (patch) | |
tree | 70cb34f9b169819b6615f3e5bc30328a14b5953c /test/test_spikes.py | |
parent | 16040c3fb9f61f7753172f6d67ac7b22c961dcaa (diff) | |
parent | fb1e0d5fd4dfc332298668a225415c5b795c0192 (diff) |
Merge pull request #31 from jonathanjouty/master
Make merge_spike_trains work with empty spike trains, and faster
Diffstat (limited to 'test/test_spikes.py')
-rw-r--r-- | test/test_spikes.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/test/test_spikes.py b/test/test_spikes.py index bcface2..ee505b5 100644 --- a/test/test_spikes.py +++ b/test/test_spikes.py @@ -85,6 +85,16 @@ def test_merge_spike_trains(): check_merged_spikes(merged_spikes.spikes, [st.spikes for st in spike_trains]) +def test_merge_empty_spike_trains(): + # first load the data + spike_trains = spk.load_spike_trains_from_txt(TEST_DATA, edges=(0, 4000)) + # take two non-empty trains, and one empty one + empty = spk.SpikeTrain([],[spike_trains[0].t_start,spike_trains[0].t_end]) + merged_spikes = spk.merge_spike_trains([spike_trains[0], empty, spike_trains[1]]) + # test if result is sorted + assert((merged_spikes.spikes == np.sort(merged_spikes.spikes)).all()) + # we don't need to check more, that's done by test_merge_spike_trains + if __name__ == "main": test_load_from_txt() |