123456789101112131415161718 |
- import numpy as np
- import pytest
- from pandas import (
- NaT,
- PeriodIndex,
- period_range,
- )
- import pandas._testing as tm
- class TestInsert:
- @pytest.mark.parametrize("na", [np.nan, NaT, None])
- def test_insert(self, na):
- # GH#18295 (test missing)
- expected = PeriodIndex(["2017Q1", NaT, "2017Q2", "2017Q3", "2017Q4"], freq="Q")
- result = period_range("2017Q1", periods=4, freq="Q").insert(1, na)
- tm.assert_index_equal(result, expected)
|