35 lines
926 B
Python
35 lines
926 B
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from pandas import (
|
||
|
|
TimedeltaIndex,
|
||
|
|
timedelta_range,
|
||
|
|
)
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
class TestRepeat:
|
||
|
|
def test_repeat(self):
|
||
|
|
index = timedelta_range("1 days", periods=2, freq="D")
|
||
|
|
exp = TimedeltaIndex(["1 days", "1 days", "2 days", "2 days"])
|
||
|
|
for res in [index.repeat(2), np.repeat(index, 2)]:
|
||
|
|
tm.assert_index_equal(res, exp)
|
||
|
|
assert res.freq is None
|
||
|
|
|
||
|
|
index = TimedeltaIndex(["1 days", "NaT", "3 days"])
|
||
|
|
exp = TimedeltaIndex(
|
||
|
|
[
|
||
|
|
"1 days",
|
||
|
|
"1 days",
|
||
|
|
"1 days",
|
||
|
|
"NaT",
|
||
|
|
"NaT",
|
||
|
|
"NaT",
|
||
|
|
"3 days",
|
||
|
|
"3 days",
|
||
|
|
"3 days",
|
||
|
|
]
|
||
|
|
)
|
||
|
|
for res in [index.repeat(3), np.repeat(index, 3)]:
|
||
|
|
tm.assert_index_equal(res, exp)
|
||
|
|
assert res.freq is None
|