diff options
author | Mario Mulansky <mario.mulansky@gmx.net> | 2017-10-04 21:56:40 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-10-04 21:56:40 -0700 |
commit | de571efd0894f0332eadec7e4a2184f9a4ffaf25 (patch) | |
tree | 0e6bf7b745a5baae0bee0a4d8a2873c8ca4bef85 /test/test_save_load.py | |
parent | f4ea757a1a32be3df440b4d742cafd5036544225 (diff) | |
parent | ecd4b5f0f7e93859c1262593e2c09e1eb6775819 (diff) |
Merge pull request #27 from mariomulansky/develop
Merge v 0.5.2
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() |