16
16
class TestSeriesQuantile (TestData , tm .TestCase ):
17
17
18
18
def test_quantile (self ):
19
- from numpy import percentile
20
19
21
20
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 ))
23
22
24
23
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 ))
26
25
27
26
# object dtype
28
27
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 ))
30
29
31
30
# datetime64[ns] dtype
32
31
dts = self .ts .index .to_series ()
@@ -48,12 +47,11 @@ def test_quantile(self):
48
47
self .ts .quantile (invalid )
49
48
50
49
def test_quantile_multi (self ):
51
- from numpy import percentile
52
50
53
51
qs = [.1 , .9 ]
54
52
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 )],
57
55
index = qs , name = self .ts .name )
58
56
tm .assert_series_equal (result , expected )
59
57
@@ -70,50 +68,44 @@ def test_quantile_multi(self):
70
68
[], dtype = float ))
71
69
tm .assert_series_equal (result , expected )
72
70
71
+ @pytest .mark .skipif (_np_version_under1p9 ,
72
+ reason = "Numpy version is under 1.9" )
73
73
def test_quantile_interpolation (self ):
74
74
# GH #10174
75
- if _np_version_under1p9 :
76
- pytest .skip ("Numpy version is under 1.9" )
77
-
78
- from numpy import percentile
79
75
80
76
# interpolation = linear (default case)
81
77
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 ))
83
79
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 ))
85
81
86
82
# test with and without interpolation keyword
87
83
self .assertEqual (q , q1 )
88
84
85
+ @pytest .mark .skipif (_np_version_under1p9 ,
86
+ reason = "Numpy version is under 1.9" )
89
87
def test_quantile_interpolation_dtype (self ):
90
88
# GH #10174
91
- if _np_version_under1p9 :
92
- pytest .skip ("Numpy version is under 1.9" )
93
-
94
- from numpy import percentile
95
89
96
90
# interpolation = linear (default case)
97
91
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 ))
99
93
self .assertTrue (is_integer (q ))
100
94
101
95
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 ))
103
97
self .assertTrue (is_integer (q ))
104
98
99
+ @pytest .mark .skipif (not _np_version_under1p9 ,
100
+ reason = "Numpy version is greater 1.9" )
105
101
def test_quantile_interpolation_np_lt_1p9 (self ):
106
102
# GH #10174
107
- if not _np_version_under1p9 :
108
- pytest .skip ("Numpy version is greater than 1.9" )
109
-
110
- from numpy import percentile
111
103
112
104
# interpolation = linear (default case)
113
105
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 ))
115
107
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 ))
117
109
118
110
# interpolation other than linear
119
111
expErrMsg = "Interpolation methods other than "
0 commit comments