Parametrize test over number of nodes to demonstrate that it is possible.

Author: heplesser

testsuite/pytests/sli2py_mpi/test_all_to_all.py

@@ -19,12 +19,16 @@
 # You should have received a copy of the GNU General Public License
 # along with NEST.  If not, see <http://www.gnu.org/licenses/>.
 
-
+import numpy as np
+import pandas
+import pytest
 from mpi_test_wrapper import MPITestAssertEqual
 
 
+# Parametrization over the number of nodes here only so show hat it works
+@pytest.mark.parametrize("N", [4, 7])
 @MPITestAssertEqual([1, 4], debug=False)
-def test_all_to_all():
+def test_all_to_all(N):
     """
     Confirm that all-to-all connections created correctly for more targets than local nodes.
     """
@@ -34,7 +38,7 @@ def test_all_to_all():
 
     nest.ResetKernel()
 
-    nrns = nest.Create("parrot_neuron", n=4)
+    nrns = nest.Create("parrot_neuron", n=N)
     nest.Connect(nrns, nrns, "all_to_all")
 
     conns = nest.GetConnections().get(output="pandas").drop(labels=["target_thread", "port"], axis=1)