Fix tests

ipython · Jan 11, 2021 · 23a6163 · 23a6163
1 parent 61fb170
commit 23a6163
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ipykernel/tests/utils.py b/ipykernel/tests/utils.py
@@ -58,7 +58,10 @@ def execute(code='', kc=None, **kwargs):
     if kc is None:
         kc = KC
     msg_id = kc.execute(code=code, **kwargs)
-    reply = kc.get_shell_msg(timeout=TIMEOUT)
+    while True:
+        reply = kc.get_shell_msg(timeout=TIMEOUT)
+        if reply['parent_header']['msg_id'] == msg_id:
+            break
     validate_message(reply, 'execute_reply', msg_id)
     busy = kc.get_iopub_msg(timeout=TIMEOUT)
     validate_message(busy, 'status', msg_id)