skips

Author: jreback

pandas/tests/series/test_quantile.py

@@ -3,6 +3,7 @@
 
 import pytest
 import numpy as np
+from numpy import percentile
 import pandas as pd
 
 from pandas import (Index, Series, _np_version_under1p9)
@@ -16,7 +17,6 @@
 class TestSeriesQuantile(TestData, tm.TestCase):
 
     def test_quantile(self):
-        from numpy import percentile
 
         q = self.ts.quantile(0.1)
         self.assertEqual(q, percentile(self.ts.valid(), 10))
@@ -48,7 +48,6 @@ def test_quantile(self):
                 self.ts.quantile(invalid)
 
     def test_quantile_multi(self):
-        from numpy import percentile
 
         qs = [.1, .9]
         result = self.ts.quantile(qs)
@@ -70,12 +69,9 @@ def test_quantile_multi(self):
             [], dtype=float))
         tm.assert_series_equal(result, expected)
 
+    @pytest.mark.skipif(_np_version_under1p9, reason='numpy < 1.9')
     def test_quantile_interpolation(self):
         # GH #10174
-        if _np_version_under1p9:
-            pytest.skip("Numpy version is under 1.9")
-
-        from numpy import percentile
 
         # interpolation = linear (default case)
         q = self.ts.quantile(0.1, interpolation='linear')
@@ -86,12 +82,9 @@ def test_quantile_interpolation(self):
         # test with and without interpolation keyword
         self.assertEqual(q, q1)
 
+    @pytest.mark.skipif(_np_version_under1p9, reason='numpy < 1.9')
     def test_quantile_interpolation_dtype(self):
         # GH #10174
-        if _np_version_under1p9:
-            pytest.skip("Numpy version is under 1.9")
-
-        from numpy import percentile
 
         # interpolation = linear (default case)
         q = pd.Series([1, 3, 4]).quantile(0.5, interpolation='lower')
@@ -102,12 +95,9 @@ def test_quantile_interpolation_dtype(self):
         self.assertEqual(q, percentile(np.array([1, 3, 4]), 50))
         self.assertTrue(is_integer(q))
 
+    @pytest.mark.skipif(not _np_version_under1p9, reason='numpy < 1.9')
     def test_quantile_interpolation_np_lt_1p9(self):
         # GH #10174
-        if not _np_version_under1p9:
-            pytest.skip("Numpy version is greater than 1.9")
-
-        from numpy import percentile
 
         # interpolation = linear (default case)
         q = self.ts.quantile(0.1, interpolation='linear')