Make split return an iterator.

pyoperators/utils/misc.py

@@ -761,13 +761,13 @@ def settingerr(*args, **keywords):
 
 def split(n, m, rank=None):
     """
-    Return an iterator through the slices that partition a list of n elements
-    in m almost same-size groups. If a rank is provided, only the slice
-    for the rank is returned.
+    Iterate through the slices that partition a list of n elements in m almost
+    same-size groups. If a rank is provided, only the slice for the rank
+    is returned.
 
     Example
     -------
-    >>> split(1000, 2)
+    >>> tuple(split(1000, 2))
     (slice(0, 500, None), slice(500, 1000, None))
     >>> split(1000, 2, 1)
     slice(500, 1000, None)
@@ -787,7 +787,7 @@ def generator():
             start += work
             rank += 1
 
-    return tuple(generator())
+    return generator()
 
 
 def strelapsed(t0, msg='Elapsed time'):

test/test_utils.py

@@ -372,7 +372,7 @@ def test_settingerr():
 
 def test_split():
     def func(n, m):
-        slices = split(n, m)
+        slices = tuple(split(n, m))
         assert_eq(len(slices), m)
         x = np.zeros(n, int)
         for s in slices: