BUGFIX: multi green thread with one item

Author: czheo

syntax_sugar/pipe.py

@@ -74,7 +74,11 @@ def multithread(fn, poolsize, data):
 
 def multigreenthread(fn, poolsize, data):
     p = GreenPool(poolsize)
-    return list(p.imap(fn, data))
+    if not hasattr(data, '__iter__'):
+        data= [data]
+        return next(p.imap(fn, data))
+    else:
+        return list(p.imap(fn, data))
 
 
 process_syntax = p = ProcessSyntax()

tests/test_pipe.py

@@ -30,12 +30,20 @@ def test_pipe_connect():
 # not all processes in the pool are necessarily used 
 # 
 def test_pipe_multiprocess():
+    assert pipe(10) | p[lambda x: x**2] | END == 10**2
+    assert pipe(10) | p[lambda x: x**2] * 2 | END == 10**2
     assert pipe(100) | range | p[lambda x: x**2] * 3 | sorted | END == [x ** 2 for x in range(100)]
 
 def test_pipe_multithread():
+    assert pipe(10) | t[lambda x: x**2] | END == 10**2
+    assert pipe(10) | t[lambda x: x**2] * 2 | END == 10**2
     assert pipe(100) | range | t[lambda x: x**2] * 3 | sorted | END == [x ** 2 for x in range(100)]
 
 def test_pipe_multigreenthread():
+    assert pipe(10) | [lambda x: x**2] | END == 10**2
+    assert pipe(10) | [lambda x: x**2] * 2 | END == 10**2
+    assert pipe(10) | g[lambda x: x**2] | END == 10**2
+    assert pipe(10) | g[lambda x: x**2] * 2 | END == 10**2
     assert pipe(10000) | range | [lambda x: x**2] * 10000 | sorted | END == [x ** 2 for x in range(10000)]
     assert pipe(10000) | range | g[lambda x: x**2] * 10000 | sorted | END == [x ** 2 for x in range(10000)]