27 lines
772 B
Python
27 lines
772 B
Python
|
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
|