projektAI/venv/Lib/site-packages/pandas/tests/indexes/period/test_fillna.py

37 lines
1.1 KiB
Python
Raw Normal View History

2021-06-06 22:13:05 +02:00
from pandas import Index, NaT, Period, PeriodIndex
import pandas._testing as tm
class TestFillNA:
def test_fillna_period(self):
# GH#11343
idx = PeriodIndex(["2011-01-01 09:00", NaT, "2011-01-01 11:00"], freq="H")
exp = PeriodIndex(
["2011-01-01 09:00", "2011-01-01 10:00", "2011-01-01 11:00"], freq="H"
)
result = idx.fillna(Period("2011-01-01 10:00", freq="H"))
tm.assert_index_equal(result, exp)
exp = Index(
[
Period("2011-01-01 09:00", freq="H"),
"x",
Period("2011-01-01 11:00", freq="H"),
],
dtype=object,
)
result = idx.fillna("x")
tm.assert_index_equal(result, exp)
exp = Index(
[
Period("2011-01-01 09:00", freq="H"),
Period("2011-01-01", freq="D"),
Period("2011-01-01 11:00", freq="H"),
],
dtype=object,
)
result = idx.fillna(Period("2011-01-01", freq="D"))
tm.assert_index_equal(result, exp)