Skip to content

Commit 6a899d3

Browse files
committed
TST: clean up series/frame api tests inheritance a bit
1 parent 2312ad6 commit 6a899d3

File tree

5 files changed

+19
-27
lines changed

5 files changed

+19
-27
lines changed
File renamed without changes.

pandas/tests/series/test_quantile.py

Lines changed: 17 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -16,17 +16,16 @@
1616
class TestSeriesQuantile(TestData, tm.TestCase):
1717

1818
def test_quantile(self):
19-
from numpy import percentile
2019

2120
q = self.ts.quantile(0.1)
22-
self.assertEqual(q, percentile(self.ts.valid(), 10))
21+
self.assertEqual(q, np.percentile(self.ts.valid(), 10))
2322

2423
q = self.ts.quantile(0.9)
25-
self.assertEqual(q, percentile(self.ts.valid(), 90))
24+
self.assertEqual(q, np.percentile(self.ts.valid(), 90))
2625

2726
# object dtype
2827
q = Series(self.ts, dtype=object).quantile(0.9)
29-
self.assertEqual(q, percentile(self.ts.valid(), 90))
28+
self.assertEqual(q, np.percentile(self.ts.valid(), 90))
3029

3130
# datetime64[ns] dtype
3231
dts = self.ts.index.to_series()
@@ -48,12 +47,11 @@ def test_quantile(self):
4847
self.ts.quantile(invalid)
4948

5049
def test_quantile_multi(self):
51-
from numpy import percentile
5250

5351
qs = [.1, .9]
5452
result = self.ts.quantile(qs)
55-
expected = pd.Series([percentile(self.ts.valid(), 10),
56-
percentile(self.ts.valid(), 90)],
53+
expected = pd.Series([np.percentile(self.ts.valid(), 10),
54+
np.percentile(self.ts.valid(), 90)],
5755
index=qs, name=self.ts.name)
5856
tm.assert_series_equal(result, expected)
5957

@@ -70,50 +68,44 @@ def test_quantile_multi(self):
7068
[], dtype=float))
7169
tm.assert_series_equal(result, expected)
7270

71+
@pytest.mark.skipif(_np_version_under1p9,
72+
reason="Numpy version is under 1.9")
7373
def test_quantile_interpolation(self):
7474
# GH #10174
75-
if _np_version_under1p9:
76-
pytest.skip("Numpy version is under 1.9")
77-
78-
from numpy import percentile
7975

8076
# interpolation = linear (default case)
8177
q = self.ts.quantile(0.1, interpolation='linear')
82-
self.assertEqual(q, percentile(self.ts.valid(), 10))
78+
self.assertEqual(q, np.percentile(self.ts.valid(), 10))
8379
q1 = self.ts.quantile(0.1)
84-
self.assertEqual(q1, percentile(self.ts.valid(), 10))
80+
self.assertEqual(q1, np.percentile(self.ts.valid(), 10))
8581

8682
# test with and without interpolation keyword
8783
self.assertEqual(q, q1)
8884

85+
@pytest.mark.skipif(_np_version_under1p9,
86+
reason="Numpy version is under 1.9")
8987
def test_quantile_interpolation_dtype(self):
9088
# GH #10174
91-
if _np_version_under1p9:
92-
pytest.skip("Numpy version is under 1.9")
93-
94-
from numpy import percentile
9589

9690
# interpolation = linear (default case)
9791
q = pd.Series([1, 3, 4]).quantile(0.5, interpolation='lower')
98-
self.assertEqual(q, percentile(np.array([1, 3, 4]), 50))
92+
self.assertEqual(q, np.percentile(np.array([1, 3, 4]), 50))
9993
self.assertTrue(is_integer(q))
10094

10195
q = pd.Series([1, 3, 4]).quantile(0.5, interpolation='higher')
102-
self.assertEqual(q, percentile(np.array([1, 3, 4]), 50))
96+
self.assertEqual(q, np.percentile(np.array([1, 3, 4]), 50))
10397
self.assertTrue(is_integer(q))
10498

99+
@pytest.mark.skipif(not _np_version_under1p9,
100+
reason="Numpy version is greater 1.9")
105101
def test_quantile_interpolation_np_lt_1p9(self):
106102
# GH #10174
107-
if not _np_version_under1p9:
108-
pytest.skip("Numpy version is greater than 1.9")
109-
110-
from numpy import percentile
111103

112104
# interpolation = linear (default case)
113105
q = self.ts.quantile(0.1, interpolation='linear')
114-
self.assertEqual(q, percentile(self.ts.valid(), 10))
106+
self.assertEqual(q, np.percentile(self.ts.valid(), 10))
115107
q1 = self.ts.quantile(0.1)
116-
self.assertEqual(q1, percentile(self.ts.valid(), 10))
108+
self.assertEqual(q1, np.percentile(self.ts.valid(), 10))
117109

118110
# interpolation other than linear
119111
expErrMsg = "Interpolation methods other than "

pandas/tests/sparse/test_frame.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
from pandas.sparse.libsparse import BlockIndex, IntIndex
2424
from pandas.sparse.api import SparseSeries, SparseDataFrame, SparseArray
25-
from pandas.tests.frame.test_misc_api import SharedWithSparse
25+
from pandas.tests.frame.test_api import SharedWithSparse
2626

2727
from pandas.tests.sparse.common import spmatrix # noqa: F401
2828

pandas/tests/sparse/test_series.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
from pandas.sparse.libsparse import BlockIndex, IntIndex
2020
from pandas.sparse.api import SparseSeries
21-
from pandas.tests.series.test_misc_api import SharedWithSparse
21+
from pandas.tests.series.test_api import SharedWithSparse
2222

2323

2424
def _test_data1():

0 commit comments

Comments
 (0)