Skip to content

Commit 1897e2f

Browse files
committed
add limit_area parameter to Series bfill and ffill methods
1 parent 8186dfd commit 1897e2f

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/test_series.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,7 +1160,13 @@ def test_types_bfill() -> None:
11601160
s1 = pd.Series([1, 2, 3])
11611161
check(assert_type(s1.bfill(), "pd.Series[int]"), pd.Series, np.integer)
11621162
check(assert_type(s1.bfill(inplace=False), "pd.Series[int]"), pd.Series, np.integer)
1163+
check(
1164+
assert_type(s1.bfill(inplace=False, limit_area="inside"), "pd.Series[int]"),
1165+
pd.Series,
1166+
np.integer,
1167+
)
11631168
assert assert_type(s1.bfill(inplace=True), None) is None
1169+
assert assert_type(s1.bfill(inplace=True, limit_area="outside"), None) is None
11641170

11651171

11661172
def test_types_ewm() -> None:
@@ -1205,7 +1211,13 @@ def test_types_ffill() -> None:
12051211
s1 = pd.Series([1, 2, 3])
12061212
check(assert_type(s1.ffill(), "pd.Series[int]"), pd.Series, np.integer)
12071213
check(assert_type(s1.ffill(inplace=False), "pd.Series[int]"), pd.Series, np.integer)
1214+
check(
1215+
assert_type(s1.ffill(inplace=False, limit_area="inside"), "pd.Series[int]"),
1216+
pd.Series,
1217+
np.integer,
1218+
)
12081219
assert assert_type(s1.ffill(inplace=True), None) is None
1220+
assert assert_type(s1.ffill(inplace=True, limit_area="outside"), None) is None
12091221

12101222

12111223
def test_types_as_type() -> None:

0 commit comments

Comments
 (0)