26 lines
853 B
Python
26 lines
853 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from pandas import Series
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
class TestStrAccessor:
|
||
|
|
def test_str_attribute(self):
|
||
|
|
# GH#9068
|
||
|
|
methods = ["strip", "rstrip", "lstrip"]
|
||
|
|
ser = Series([" jack", "jill ", " jesse ", "frank"])
|
||
|
|
for method in methods:
|
||
|
|
expected = Series([getattr(str, method)(x) for x in ser.values])
|
||
|
|
tm.assert_series_equal(getattr(Series.str, method)(ser.str), expected)
|
||
|
|
|
||
|
|
# str accessor only valid with string values
|
||
|
|
ser = Series(range(5))
|
||
|
|
with pytest.raises(AttributeError, match="only use .str accessor"):
|
||
|
|
ser.str.repeat(2)
|
||
|
|
|
||
|
|
def test_str_accessor_updates_on_inplace(self):
|
||
|
|
ser = Series(list("abc"))
|
||
|
|
return_value = ser.drop([0], inplace=True)
|
||
|
|
assert return_value is None
|
||
|
|
assert len(ser.str.lower()) == 2
|