74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from pandas import (
|
||
|
|
Index,
|
||
|
|
Series,
|
||
|
|
date_range,
|
||
|
|
)
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
class TestSeriesDelItem:
|
||
|
|
def test_delitem(self):
|
||
|
|
# GH#5542
|
||
|
|
# should delete the item inplace
|
||
|
|
s = Series(range(5))
|
||
|
|
del s[0]
|
||
|
|
|
||
|
|
expected = Series(range(1, 5), index=range(1, 5))
|
||
|
|
tm.assert_series_equal(s, expected)
|
||
|
|
|
||
|
|
del s[1]
|
||
|
|
expected = Series(range(2, 5), index=range(2, 5))
|
||
|
|
tm.assert_series_equal(s, expected)
|
||
|
|
|
||
|
|
# only 1 left, del, add, del
|
||
|
|
s = Series(1)
|
||
|
|
del s[0]
|
||
|
|
tm.assert_series_equal(s, Series(dtype="int64", index=Index([], dtype="int64")))
|
||
|
|
s[0] = 1
|
||
|
|
tm.assert_series_equal(s, Series(1))
|
||
|
|
del s[0]
|
||
|
|
tm.assert_series_equal(s, Series(dtype="int64", index=Index([], dtype="int64")))
|
||
|
|
|
||
|
|
def test_delitem_object_index(self):
|
||
|
|
# Index(dtype=object)
|
||
|
|
s = Series(1, index=["a"])
|
||
|
|
del s["a"]
|
||
|
|
tm.assert_series_equal(
|
||
|
|
s, Series(dtype="int64", index=Index([], dtype="object"))
|
||
|
|
)
|
||
|
|
s["a"] = 1
|
||
|
|
tm.assert_series_equal(s, Series(1, index=["a"]))
|
||
|
|
del s["a"]
|
||
|
|
tm.assert_series_equal(
|
||
|
|
s, Series(dtype="int64", index=Index([], dtype="object"))
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_delitem_missing_key(self):
|
||
|
|
# empty
|
||
|
|
s = Series(dtype=object)
|
||
|
|
|
||
|
|
with pytest.raises(KeyError, match=r"^0$"):
|
||
|
|
del s[0]
|
||
|
|
|
||
|
|
def test_delitem_extension_dtype(self):
|
||
|
|
# GH#40386
|
||
|
|
# DatetimeTZDtype
|
||
|
|
dti = date_range("2016-01-01", periods=3, tz="US/Pacific")
|
||
|
|
ser = Series(dti)
|
||
|
|
|
||
|
|
expected = ser[[0, 2]]
|
||
|
|
del ser[1]
|
||
|
|
assert ser.dtype == dti.dtype
|
||
|
|
tm.assert_series_equal(ser, expected)
|
||
|
|
|
||
|
|
# PeriodDtype
|
||
|
|
pi = dti.tz_localize(None).to_period("D")
|
||
|
|
ser = Series(pi)
|
||
|
|
|
||
|
|
expected = ser[:2]
|
||
|
|
del ser[2]
|
||
|
|
assert ser.dtype == pi.dtype
|
||
|
|
tm.assert_series_equal(ser, expected)
|