Fix expected text depending on IPython version. (#1354)

ipython · Feb 24, 2025 · 4854752 · 4854752
1 parent b3a0999
commit 4854752
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_start_kernel.py b/tests/test_start_kernel.py
@@ -14,6 +14,14 @@
 
 @flaky(max_runs=3)
 def test_ipython_start_kernel_userns():
+    import IPython
+
+    if IPython.version_info > (9, 0):  # noqa:SIM108
+        EXPECTED = "IPythonMainModule"
+    else:
+        # not this since https://github.com/ipython/ipython/pull/14754
+        EXPECTED = "DummyMod"
+
     cmd = dedent(
         """
         from ipykernel.kernelapp import launch_new_instance
@@ -40,7 +48,7 @@ def test_ipython_start_kernel_userns():
         content = msg["content"]
         assert content["found"]
         text = content["data"]["text/plain"]
-        assert "DummyMod" in text
+        assert EXPECTED in text
 
 
 @flaky(max_runs=3)