Skip to content

Commit c41ec61

Browse files
committed
feat: test future_stack only for LTE 2.2
1 parent abfb17e commit c41ec61

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

tests/test_frame.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2256,20 +2256,21 @@ def test_frame_stack() -> None:
22562256
),
22572257
pd.Series,
22582258
)
2259-
check(
2260-
assert_type(
2261-
df_multi_level_cols2.stack(0, future_stack=False),
2262-
Union[pd.DataFrame, "pd.Series[Any]"],
2263-
),
2264-
pd.DataFrame,
2265-
)
22662259
check(
22672260
assert_type(
22682261
df_multi_level_cols2.stack(0, dropna=True, sort=True),
22692262
Union[pd.DataFrame, "pd.Series[Any]"],
22702263
),
22712264
pd.DataFrame,
22722265
)
2266+
if PD_LTE_22:
2267+
check(
2268+
assert_type(
2269+
df_multi_level_cols2.stack(0, future_stack=False),
2270+
Union[pd.DataFrame, "pd.Series[Any]"],
2271+
),
2272+
pd.DataFrame,
2273+
)
22732274

22742275

22752276
def test_frame_reindex() -> None:

0 commit comments

Comments
 (0)