Fixed worker_pids test case.

binarymatt · Jan 22, 2013 · e119d7e · e119d7e
1 parent 062b548
commit e119d7e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_worker.py b/tests/test_worker.py
@@ -311,13 +311,13 @@ def test_worker_pids(self):
             os.waitpid(pid, 0)
 
         # ensure worker_pids() returned the correct pids
-        assert len(worker_pids) == len(pids)
         for pid in pids:
             assert str(pid) in worker_pids
 
-        # ensure there are no longer any workers running
+        # ensure the workers are no longer returned by worker_pids()
         worker_pids = Worker.worker_pids()
-        assert len(worker_pids) == 0
+        for pid in pids:
+            assert str(pid) not in worker_pids
 
     def spawn_worker(self, queues):
         pid = os.fork()