From f057e34ca084d78a655926f30d5df43e673714fb Mon Sep 17 00:00:00 2001 From: Mario Mulansky Date: Thu, 24 Mar 2016 15:56:41 +0100 Subject: added test for save spike train functionality --- test/test_save_load.py | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 test/test_save_load.py 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 + +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() -- cgit v1.2.3