3 files changed, 34 insertions, 0 deletions
diff --git a/tests/simple_wait_end_3n/node.py b/tests/simple_wait_end_3n/node.py
new file mode 100644
index 0000000..a8fbcd7
--- /dev/null
+++ b/tests/simple_wait_end_3n/node.py
@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+
+def execute(api):
+    wait=api.node_id
+    api.log("Before wait for "+str(wait)+"s")
+    api.wait(wait) # Since 3 nodes max(wait)==2
+
+    # Ensure that wait end return back when simulation ends
+    api.wait_end()
+    api.log("Terminated")
diff --git a/tests/simple_wait_end_3n/out b/tests/simple_wait_end_3n/out
new file mode 100644
index 0000000..321b0c2
--- /dev/null
+++ b/tests/simple_wait_end_3n/out
@@ -0,0 +1,7 @@
+[t=0.000,src=n0] Before wait for 0s
+[t=0.000,src=n1] Before wait for 1s
+[t=0.000,src=n2] Before wait for 2s
+[t=2.000,src=n0] Terminated
+[t=2.000,src=n1] Terminated
+[t=2.000,src=n2] Terminated
+[t=2.000,src=esds] Simulation ends
diff --git a/tests/simple_wait_end_3n/simulator.py b/tests/simple_wait_end_3n/simulator.py
new file mode 100755
index 0000000..d4c9ec4
--- /dev/null
+++ b/tests/simple_wait_end_3n/simulator.py
@@ -0,0 +1,17 @@
+#!/usr/bin/env python
+
+# Load ESDS
+import sys
+sys.path.append("../../")
+import esds
+import numpy as np
+
+B=np.full((3,3),2)
+L=np.full((3,3),0)
+s=esds.Simulator(B,L)
+
+s.create_node("node")
+s.create_node("node")
+s.create_node("node")
+
+s.run()