Merge pull request #907 from minrk/raises-remote

include traceback in wrong error in raises_remote

Author: minrk

ipyparallel/engine/kernel.py

@@ -139,12 +139,10 @@ def do_apply(self, content, bufs, msg_id, reply_metadata):
         try:
             working = shell.user_ns
 
-            prefix = "_" + str(msg_id).replace("-", "") + "_"
+            prefix = f"_{str(msg_id).replace('-', '_')}_"
 
             f, args, kwargs = unpack_apply_message(bufs, working, copy=False)
 
-            fname = getattr(f, '__name__', 'f')
-
             fname = prefix + "f"
             argname = prefix + "args"
             kwargname = prefix + "kwargs"
@@ -153,13 +151,18 @@ def do_apply(self, content, bufs, msg_id, reply_metadata):
             ns = {fname: f, argname: args, kwargname: kwargs, resultname: None}
             # print ns
             working.update(ns)
-            code = f"{resultname} = {fname}(*{argname},**{kwargname})"
+            code = f"{resultname} = {fname}(*{argname}, **{kwargname})"
             try:
                 exec(code, shell.user_global_ns, shell.user_ns)
                 result = working.get(resultname)
             finally:
                 for key in ns:
-                    working.pop(key)
+                    try:
+                        working.pop(key)
+                    except KeyError:
+                        self.log.warning(
+                            f"Failed to undefine temporary apply variable {key}, already undefined"
+                        )
 
             result_buf = serialize_object(
                 result,

ipyparallel/tests/clienttest.py

@@ -106,9 +106,10 @@ def raises_remote(etype):
         except error.CompositeError as e:
             e.raise_exception()
     except error.RemoteError as e:
+        tb = '\n'.join(e.render_traceback())
         assert (
             expected_ename == e.ename
-        ), f"Should have raised {expected_ename}, but raised {e.ename}"
+        ), f"Should have raised {expected_ename}, but raised {e.ename}:\n{tb}"
 
     else:
         pytest.fail("should have raised a RemoteError")