don't test old-timey pylab

test matplotlib directly

Author: minrk

Diff for: ipyparallel/tests/test_magics.py

@@ -440,24 +440,28 @@ def test_result(self):
                 ip.run_line_magic('pxresult', '')
             assert str(data[name]) in io.stdout
 
-    def test_px_pylab(self):
-        """%pylab works on engines"""
+    def test_px_matplotlib(self):
+        """%matplotlib inline works on engines"""
         pytest.importorskip('matplotlib')
         ip = get_ipython()
         v = self.client[-1]
         v.block = True
         v.activate()
 
         with capture_output() as io:
-            ip.run_line_magic("px", "%pylab inline")
-
-        assert (
-            "Populating the interactive namespace from numpy and matplotlib"
-            in io.stdout
-        )
+            ip.run_line_magic(
+                "px",
+                "\n".join(
+                    [
+                        "%matplotlib inline",
+                        "import numpy as np",
+                        "import matplotlib.pyplot as plt",
+                    ]
+                ),
+            )
 
         with capture_output(display=False) as io:
-            ip.run_line_magic("px", "plot(rand(100))")
+            ip.run_line_magic("px", "plt.plot(np.rand(100))")
         assert 'Out[' in io.stdout
         assert 'matplotlib.lines' in io.stdout