misc IPython 9 compatibility

ipyparallel/client/client.py

@@ -49,6 +49,7 @@
 from ipyparallel import error, serialize, util
 from ipyparallel.serialize import PrePickled, Reference
 from ipyparallel.util import _OutputProducingThread as Thread
+from ipyparallel.util import _TermColors
 
 from .asyncresult import AsyncHubResult, AsyncResult
 from .futures import MessageFuture, multi_future
@@ -161,11 +162,30 @@ def _plaintext(self) -> str:
         if not text_out:
             return ''
 
+        ip = get_ipython()
+        if ip is None:
+            colors = "NoColor"
+        else:
+            colors = ip.colors
+
+        if colors == "NoColor":
+            out = normal = ""
+        else:
+            out = _TermColors.Red
+            normal = _TermColors.Normal
+
         if '\n' in text_out and not text_out.startswith('\n'):
             # add newline for multiline reprs
             text_out = '\n' + text_out
 
-        return f"Out[{self.metadata['engine_id']}:{self.execution_count}]: {text_out}"
+        return ''.join(
+            [
+                out,
+                f"Out[{self.metadata['engine_id']}:{self.execution_count}]: ",
+                normal,
+                text_out,
+            ]
+        )
 
     def _repr_pretty_(self, p, cycle):
         p.text(self._plaintext())

ipyparallel/tests/clienttest.py

@@ -55,7 +55,7 @@ def generate_output():
     a rich displayable object.
     """
 
-    from IPython.core.display import HTML, Math, display
+    from IPython.display import HTML, Math, display
 
     print("stdout")
     print("stderr", file=sys.stderr)

ipyparallel/tests/test_magics.py

@@ -173,18 +173,18 @@ def test_cellpx_groupby_order(self):
         expected.extend(
             [
                 r'\[output:\d+\]',
-                'IPython.core.display.HTML',
+                'IPython..+.HTML',
             ]
             * len(v)
         )
         expected.extend(
             [
                 r'\[output:\d+\]',
-                'IPython.core.display.Math',
+                'IPython..+.Math',
             ]
             * len(v)
         )
-        expected.extend([r'Out\[\d+:\d+\]:.*IPython\.core\.display\.Math'] * len(v))
+        expected.extend([r'Out\[\d+:\d+\]:.*IPython\..+\.Math'] * len(v))
 
         assert len(lines), len(expected) == io.stdout
         for line, expect in zip(lines, expected):

ipyparallel/tests/test_view.py

@@ -614,7 +614,7 @@ def test_execute_magic(self):
     def test_execute_displaypub(self):
         """execute tracks display_pub output"""
         view = self.client[:]
-        view.execute("from IPython.core.display import *")
+        view.execute("from IPython.display import *")
         ar = view.execute("[ display(i) for i in range(5) ]", block=True)
 
         expected = [{'text/plain': str(j)} for j in range(5)]
@@ -625,7 +625,7 @@ def test_execute_displaypub(self):
     def test_apply_displaypub(self):
         """apply tracks display_pub output"""
         view = self.client[:]
-        view.execute("from IPython.core.display import *")
+        view.execute("from IPython.display import *")
 
         @interactive
         def publish():

ipyparallel/util.py

@@ -836,3 +836,10 @@ def __init__(self, target, **kwargs):
     def _wrapped_target(self, target, *args, **kwargs):
         _detach_thread_output(self.ident)
         return target(*args, **kwargs)
+
+
+# minimal subset of TermColors, removed from IPython
+# not for public consumption
+class _TermColors:
+    Normal = '\033[0m'
+    Red = '\033[0;31m'