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
| import numpy as np
| import pytest
|
| from pandas import (
| PeriodIndex,
| period_range,
| )
| import pandas._testing as tm
|
|
| class TestRepeat:
| @pytest.mark.parametrize("use_numpy", [True, False])
| @pytest.mark.parametrize(
| "index",
| [
| period_range("2000-01-01", periods=3, freq="D"),
| period_range("2001-01-01", periods=3, freq="2D"),
| PeriodIndex(["2001-01", "NaT", "2003-01"], freq="M"),
| ],
| )
| def test_repeat_freqstr(self, index, use_numpy):
| # GH#10183
| expected = PeriodIndex([per for per in index for _ in range(3)])
| result = np.repeat(index, 3) if use_numpy else index.repeat(3)
| tm.assert_index_equal(result, expected)
| assert result.freqstr == index.freqstr
|
|
