fixing up small pytest issues

pizzathief · pizzathief · commit 78587c10d887 · 2016-01-21T12:09:06.000+11:00
diff --git a/lib/matplotlib/tests/test_mlab.py b/lib/matplotlib/tests/test_mlab.py
@@ -2751,7 +2751,7 @@ def get_z(x, y):
                                   np.ma.getmask(correct_zi_masked))
 
 
-@pytest.mark.xfail(not HAS_NATGRID, 
+@pytest.mark.xfail(not HAS_NATGRID,
                    reason='import of the natgrid toolkit failed')
 def test_griddata_nn():
     # z is a linear function of x and y.
diff --git a/lib/matplotlib/tests/test_patheffects.py b/lib/matplotlib/tests/test_patheffects.py
@@ -8,6 +8,7 @@
 from matplotlib.testing.decorators import (image_comparison, cleanup)
 import matplotlib.pyplot as plt
 import matplotlib.patheffects as path_effects
+import pytest
 
 try:
     # mock in python 3.3+
