summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMario Mulansky <mario.mulansky@gmx.net>2016-03-24 15:56:41 +0100
committerMario Mulansky <mario.mulansky@gmx.net>2016-03-24 15:56:41 +0100
commitf057e34ca084d78a655926f30d5df43e673714fb (patch)
treed0387333d09e21de7e7f67e8f61ada313225e415
parent0d8af2c97d766a4fa514f0232189bc17c31c67a0 (diff)
added test for save spike train functionality
-rw-r--r--test/test_save_load.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/test/test_save_load.py b/test/test_save_load.py
new file mode 100644
index 0000000..ff222d3
--- /dev/null
+++ b/test/test_save_load.py
@@ -0,0 +1,42 @@
+""" test_save_load.py
+
+Tests saving and loading of spike trains
+
+Copyright 2016, Mario Mulansky <mario.mulansky@gmx.net>
+
+Distributed under the BSD License
+
+"""
+
+from __future__ import print_function
+from numpy.testing import assert_array_equal
+
+import tempfile
+import os.path
+
+import pyspike as spk
+
+
+def test_save_load():
+ file_name = os.path.join(tempfile.mkdtemp(prefix='pyspike_'),
+ "save_load.txt")
+
+ N = 10
+ # generate some spike trains
+ spike_trains = []
+ for n in xrange(N):
+ spike_trains.append(spk.generate_poisson_spikes(1.0, [0, 100]))
+
+ # save them into txt file
+ spk.save_spike_trains_to_txt(spike_trains, file_name, precision=17)
+
+ # load again
+ spike_trains_loaded = spk.load_spike_trains_from_txt(file_name, [0, 100])
+
+ for n in xrange(N):
+ assert_array_equal(spike_trains[n].spikes,
+ spike_trains_loaded[n].spikes)
+
+
+if __name__ == "__main__":
+ test_save_load()