You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
853 B
25 lines
853 B
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
|
|
|