allow map / parallel function for single-engine views

ivanov · Apr 16, 2012 · b1f1d1e · b1f1d1e
1 parent f57191a
commit b1f1d1e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/IPython/parallel/client/remotefunction.py b/IPython/parallel/client/remotefunction.py
@@ -182,6 +182,9 @@ def __call__(self, *sequences):
             # 'all' is lazily evaluated at execution time, which is now:
             if targets == 'all':
                 targets = client._build_targets(targets)[1]
+            elif isinstance(targets, int):
+                # single-engine view, targets must be iterable
+                targets = [targets]
             nparts = len(targets)
 
         msg_ids = []

diff --git a/IPython/parallel/tests/test_view.py b/IPython/parallel/tests/test_view.py
@@ -534,4 +534,10 @@ def test_reference_nameerror(self):
         echo = lambda x:x
         self.assertRaisesRemote(NameError, v.apply_sync, echo, r)
 
+    def test_single_engine_map(self):
+        e0 = self.client[self.client.ids[0]]
+        r = range(5)
+        check = [ -1*i for i in r ]
+        result = e0.map_sync(lambda x: -1*x, r)
+        self.assertEquals(result, check)