19 lines
482 B
Python
19 lines
482 B
Python
|
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)
|