test_fillna.py 597 B

12345678910111213141516171819202122
  1. from pandas import (
  2. Index,
  3. NaT,
  4. Timedelta,
  5. TimedeltaIndex,
  6. )
  7. import pandas._testing as tm
  8. class TestFillNA:
  9. def test_fillna_timedelta(self):
  10. # GH#11343
  11. idx = TimedeltaIndex(["1 day", NaT, "3 day"])
  12. exp = TimedeltaIndex(["1 day", "2 day", "3 day"])
  13. tm.assert_index_equal(idx.fillna(Timedelta("2 day")), exp)
  14. exp = TimedeltaIndex(["1 day", "3 hour", "3 day"])
  15. idx.fillna(Timedelta("3 hour"))
  16. exp = Index([Timedelta("1 day"), "x", Timedelta("3 day")], dtype=object)
  17. tm.assert_index_equal(idx.fillna("x"), exp)