1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
| 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
|
|