diff options
author | Mario Mulansky <mario.mulansky@gmx.net> | 2017-10-04 22:01:52 -0700 |
---|---|---|
committer | Mario Mulansky <mario.mulansky@gmx.net> | 2017-10-04 22:01:52 -0700 |
commit | 2af020ee53a006436fb69bf80b28d5a850ae8c43 (patch) | |
tree | 988d0be91bc678169fdc7934d5494dcf9f8f8768 /test/test_save_load.py | |
parent | 9fa10fe30e3216170f41ce47797c7a3a56e17a5f (diff) | |
parent | de571efd0894f0332eadec7e4a2184f9a4ffaf25 (diff) |
Merge branch 'master' of github.com:mariomulansky/PySpike
Diffstat (limited to 'test/test_save_load.py')
-rw-r--r-- | test/test_save_load.py | 42 |
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..33249b3 --- /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 range(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 range(N): + assert_array_equal(spike_trains[n].spikes, + spike_trains_loaded[n].spikes) + + +if __name__ == "__main__": + test_save_load() |