36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from pandas import Series, bdate_range, date_range, period_range
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
class TestBetween:
|
||
|
|
|
||
|
|
# TODO: redundant with test_between_datetime_values?
|
||
|
|
def test_between(self):
|
||
|
|
series = Series(date_range("1/1/2000", periods=10))
|
||
|
|
left, right = series[[2, 7]]
|
||
|
|
|
||
|
|
result = series.between(left, right)
|
||
|
|
expected = (series >= left) & (series <= right)
|
||
|
|
tm.assert_series_equal(result, expected)
|
||
|
|
|
||
|
|
def test_between_datetime_values(self):
|
||
|
|
ser = Series(bdate_range("1/1/2000", periods=20).astype(object))
|
||
|
|
ser[::2] = np.nan
|
||
|
|
|
||
|
|
result = ser[ser.between(ser[3], ser[17])]
|
||
|
|
expected = ser[3:18].dropna()
|
||
|
|
tm.assert_series_equal(result, expected)
|
||
|
|
|
||
|
|
result = ser[ser.between(ser[3], ser[17], inclusive=False)]
|
||
|
|
expected = ser[5:16].dropna()
|
||
|
|
tm.assert_series_equal(result, expected)
|
||
|
|
|
||
|
|
def test_between_period_values(self):
|
||
|
|
ser = Series(period_range("2000-01-01", periods=10, freq="D"))
|
||
|
|
left, right = ser[[2, 7]]
|
||
|
|
result = ser.between(left, right)
|
||
|
|
expected = (ser >= left) & (ser <= right)
|
||
|
|
tm.assert_series_equal(result, expected)
|