1
1
import pytest
2
2
import numpy as np
3
- import pandas as pd
4
3
5
4
from pandas import Series , IntervalIndex , Interval
6
5
import pandas .util .testing as tm
@@ -170,17 +169,17 @@ def test_loc_with_overlap(self):
170
169
171
170
# interval
172
171
expected = 0
173
- result = s .loc [pd . interval (1 , 5 )]
172
+ result = s .loc [Interval (1 , 5 )]
174
173
tm .assert_series_equal (expected , result )
175
174
176
- result = s [pd . interval (1 , 5 )]
175
+ result = s [Interval (1 , 5 )]
177
176
tm .assert_series_equal (expected , result )
178
177
179
178
expected = s
180
- result = s .loc [[pd . interval (1 , 5 ), pd . Interval (3 , 7 )]]
179
+ result = s .loc [[Interval (1 , 5 ), Interval (3 , 7 )]]
181
180
tm .assert_series_equal (expected , result )
182
181
183
- result = s [[pd . interval (1 , 5 ), pd . Interval (3 , 7 )]]
182
+ result = s [[Interval (1 , 5 ), Interval (3 , 7 )]]
184
183
tm .assert_series_equal (expected , result )
185
184
186
185
with pytest .raises (KeyError ):
@@ -197,17 +196,17 @@ def test_loc_with_overlap(self):
197
196
198
197
# slices with interval (only exact matches)
199
198
expected = s
200
- result = s .loc [pd . interval (1 , 5 ):pd . Interval (3 , 7 )]
199
+ result = s .loc [Interval (1 , 5 ):Interval (3 , 7 )]
201
200
tm .assert_series_equal (expected , result )
202
201
203
- result = s [pd . interval (1 , 5 ):pd . Interval (3 , 7 )]
202
+ result = s [Interval (1 , 5 ):Interval (3 , 7 )]
204
203
tm .assert_series_equal (expected , result )
205
204
206
205
with pytest .raises (KeyError ):
207
- s .loc [pd . interval (1 , 6 ):pd . Interval (3 , 8 )]
206
+ s .loc [Interval (1 , 6 ):Interval (3 , 8 )]
208
207
209
208
with pytest .raises (KeyError ):
210
- s [pd . interval (1 , 6 ):pd . Interval (3 , 8 )]
209
+ s [Interval (1 , 6 ):Interval (3 , 8 )]
211
210
212
211
# slices with scalar raise for overlapping intervals
213
212
# TODO KeyError is the appropriate error?
@@ -217,7 +216,7 @@ def test_loc_with_overlap(self):
217
216
def test_non_unique (self ):
218
217
219
218
idx = IntervalIndex .from_tuples ([(1 , 3 ), (3 , 7 )])
220
- s = pd . Series (range (len (idx )), index = idx )
219
+ s = Series (range (len (idx )), index = idx )
221
220
222
221
result = s .loc [Interval (1 , 3 )]
223
222
assert result == 0
0 commit comments