Skip to content

Commit 142048c

Browse files
SaturnFromTitanMateusz Górski
authored and
Mateusz Górski
committed
[pandas-dev#22550] Remove TestData from series-tests test_quantile.py (pandas-dev#29096)
1 parent b206f3e commit 142048c

File tree

1 file changed

+27
-24
lines changed

1 file changed

+27
-24
lines changed

pandas/tests/series/test_quantile.py

+27-24
Original file line numberDiff line numberDiff line change
@@ -8,24 +8,22 @@
88
from pandas.core.indexes.datetimes import Timestamp
99
import pandas.util.testing as tm
1010

11-
from .common import TestData
1211

12+
class TestSeriesQuantile:
13+
def test_quantile(self, datetime_series):
1314

14-
class TestSeriesQuantile(TestData):
15-
def test_quantile(self):
15+
q = datetime_series.quantile(0.1)
16+
assert q == np.percentile(datetime_series.dropna(), 10)
1617

17-
q = self.ts.quantile(0.1)
18-
assert q == np.percentile(self.ts.dropna(), 10)
19-
20-
q = self.ts.quantile(0.9)
21-
assert q == np.percentile(self.ts.dropna(), 90)
18+
q = datetime_series.quantile(0.9)
19+
assert q == np.percentile(datetime_series.dropna(), 90)
2220

2321
# object dtype
24-
q = Series(self.ts, dtype=object).quantile(0.9)
25-
assert q == np.percentile(self.ts.dropna(), 90)
22+
q = Series(datetime_series, dtype=object).quantile(0.9)
23+
assert q == np.percentile(datetime_series.dropna(), 90)
2624

2725
# datetime64[ns] dtype
28-
dts = self.ts.index.to_series()
26+
dts = datetime_series.index.to_series()
2927
q = dts.quantile(0.2)
3028
assert q == Timestamp("2000-01-10 19:12:00")
3129

@@ -41,20 +39,23 @@ def test_quantile(self):
4139
msg = "percentiles should all be in the interval \\[0, 1\\]"
4240
for invalid in [-1, 2, [0.5, -1], [0.5, 2]]:
4341
with pytest.raises(ValueError, match=msg):
44-
self.ts.quantile(invalid)
42+
datetime_series.quantile(invalid)
4543

46-
def test_quantile_multi(self):
44+
def test_quantile_multi(self, datetime_series):
4745

4846
qs = [0.1, 0.9]
49-
result = self.ts.quantile(qs)
47+
result = datetime_series.quantile(qs)
5048
expected = pd.Series(
51-
[np.percentile(self.ts.dropna(), 10), np.percentile(self.ts.dropna(), 90)],
49+
[
50+
np.percentile(datetime_series.dropna(), 10),
51+
np.percentile(datetime_series.dropna(), 90),
52+
],
5253
index=qs,
53-
name=self.ts.name,
54+
name=datetime_series.name,
5455
)
5556
tm.assert_series_equal(result, expected)
5657

57-
dts = self.ts.index.to_series()
58+
dts = datetime_series.index.to_series()
5859
dts.name = "xxx"
5960
result = dts.quantile((0.2, 0.2))
6061
expected = Series(
@@ -64,18 +65,20 @@ def test_quantile_multi(self):
6465
)
6566
tm.assert_series_equal(result, expected)
6667

67-
result = self.ts.quantile([])
68-
expected = pd.Series([], name=self.ts.name, index=Index([], dtype=float))
68+
result = datetime_series.quantile([])
69+
expected = pd.Series(
70+
[], name=datetime_series.name, index=Index([], dtype=float)
71+
)
6972
tm.assert_series_equal(result, expected)
7073

71-
def test_quantile_interpolation(self):
74+
def test_quantile_interpolation(self, datetime_series):
7275
# see gh-10174
7376

7477
# interpolation = linear (default case)
75-
q = self.ts.quantile(0.1, interpolation="linear")
76-
assert q == np.percentile(self.ts.dropna(), 10)
77-
q1 = self.ts.quantile(0.1)
78-
assert q1 == np.percentile(self.ts.dropna(), 10)
78+
q = datetime_series.quantile(0.1, interpolation="linear")
79+
assert q == np.percentile(datetime_series.dropna(), 10)
80+
q1 = datetime_series.quantile(0.1)
81+
assert q1 == np.percentile(datetime_series.dropna(), 10)
7982

8083
# test with and without interpolation keyword
8184
assert q == q1

0 commit comments

Comments
 (0)