added merge spikes test

1 files changed, 49 insertions, 0 deletions

diff --git a/test/test_merge_spikes.py b/test/test_merge_spikes.py
new file mode 100644
index 0000000..3162700
--- /dev/null
+++ b/test/test_merge_spikes.py
@@ -0,0 +1,49 @@
+""" test_merge_spikes.py
+
+Tests merging spikes
+
+Copyright 2014, Mario Mulansky <mario.mulansky@gmx.net>
+"""
+from __future__ import print_function
+import numpy as np
+
+import pyspike as spk
+
+def check_merged_spikes( merged_spikes, spike_trains ):
+    # create a flat array with all spike events
+    all_spikes = np.array([])
+    for spike_train in spike_trains:
+        all_spikes = np.append(all_spikes, spike_train)
+    indices = np.zeros_like(all_spikes, dtype='bool')
+    # check if we find all the spike events in the original spike trains
+    for x in merged_spikes:
+        i = np.where(all_spikes == x)[0][0] # the first axis and the first entry
+        # change to something impossible so we dont find this event again
+        all_spikes[i] = -1.0
+        indices[i] = True
+    assert( indices.all() )
+
+def test_merge_spike_trains():
+
+    # first load the data
+    spike_trains = []
+    spike_file = open("SPIKY_testdata.txt", 'r')
+    for line in spike_file:
+        spike_trains.append(spk.spike_train_from_string(line))
+        
+    spikes = spk.merge_spike_trains([spike_trains[0], spike_trains[1]])
+    # test if result is sorted
+    assert((spikes == np.sort(spikes)).all())
+    # check merging
+    check_merged_spikes( spikes, [spike_trains[0], spike_trains[1]] )
+
+    spikes = spk.merge_spike_trains(spike_trains)
+    # test if result is sorted
+    assert((spikes == np.sort(spikes)).all())
+    # check merging
+    check_merged_spikes( spikes, spike_trains )
+
+
+if __name__ == "main":
+    test_merge_spike_trains()
+