fix queue test

usc-isi-i2 · Jul 14, 2020 · a78559f · a78559f
1 parent 88fc0b6
commit a78559f
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/pyrallel/tests/test_queue.py b/pyrallel/tests/test_queue.py
@@ -4,17 +4,18 @@
 
 
 def f(iq, oq):
-    while True:
-        try:
+    try:
+        while True:
             oq.put(iq.get(timeout=2))
-        except queue.Empty:
-            return
+    except queue.Empty:
+        return
 
 
 def test_shmqueue():
     if not hasattr(pyrallel, 'ShmQueue'):
         return
 
+    mp.set_start_method('fork')
     ShmQueueCls = getattr(pyrallel, 'ShmQueue')
     sq = ShmQueueCls(chunk_size=1024 * 4, maxsize=5)
     q = mp.Queue()
@@ -26,7 +27,7 @@ def test_shmqueue():
     items = list(range(10))
 
     for i in items:
-        q.put(i)
+        sq.put(i)
 
     while True:
         try:
@@ -37,4 +38,5 @@ def test_shmqueue():
 
     p1.join()
     p2.join()
+    sq.close()
     q.close()