Source code for nest.tests.test_dataconnect
# -*- coding: utf-8 -*-
#
# test_dataconnect.py
#
# This file is part of NEST.
#
# Copyright (C) 2004 The NEST Initiative
#
# NEST is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
#
# NEST is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with NEST. If not, see <http://www.gnu.org/licenses/>.
"""
DataConnect
"""
import unittest
import nest
[docs]@nest.check_stack
class DataConnectTestCase(unittest.TestCase):
"""Find connections and test if values can be set."""
[docs] def test_DataConnect(self):
"""DataConnect"""
nest.ResetKernel()
a = nest.Create("iaf_psc_alpha", 10)
sources = [1]
target = [1.0 * x for x in range(2, 10)]
weight = [2.0 * x for x in target]
delay = [1.0 * x for x in target]
connections = [{'target': target, 'weight': weight,
'delay': delay} for t in target]
nest.DataConnect(sources, connections)
conn1 = nest.GetConnections(sources)
stat1 = nest.GetStatus(conn1)
target1 = [d['target'] for d in stat1]
self.assertEqual(target, target1)
# Now we test the connectome version of data-connect
# by erasing and reinstantiating
# the nestwork from the status data.
nest.ResetKernel()
a = nest.Create("iaf_psc_alpha", 10)
nest.DataConnect(stat1)
conn2 = nest.GetConnections()
c1 = [list(x) for x in conn1]
c2 = [list(x) for x in conn2]
self.assertEqual(c1, c2)
[docs]def suite():
suite = unittest.makeSuite(DataConnectTestCase, 'test')
return suite
[docs]def run():
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite())
if __name__ == "__main__":
run()