From b6dce27d75e1077911eab52dd639659d60a73cb1 2013-11-22 19:17:36
From: Min RK <benjaminrk@gmail.com>
Date: 2013-11-22 19:17:36
Subject: [PATCH] Merge pull request #4568 from minrk/wait-fix-again-keyerror

don't use lazily-evaluated rc.ids in wait_for_idle
---

diff --git a/IPython/parallel/tests/test_client.py b/IPython/parallel/tests/test_client.py
index 0422739..3e4a0d6 100644
--- a/IPython/parallel/tests/test_client.py
+++ b/IPython/parallel/tests/test_client.py
@@ -325,7 +325,7 @@ class TestClient(ClusterTestCase):
         # timeout 5s, polling every 100ms
         qs = rc.queue_status()
         for i in range(50):
-            if qs['unassigned'] or any(qs[eid]['tasks'] + qs[eid]['queue'] for eid in rc.ids):
+            if qs['unassigned'] or any(qs[eid]['tasks'] + qs[eid]['queue'] for eid in qs if eid != 'unassigned'):
                 time.sleep(0.1)
                 qs = rc.queue_status()
             else: