Skip to content

Commit 3ca64e2

Browse files
Add types to assert_frame_equal
1 parent 0de9955 commit 3ca64e2

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

pandas/util/testing.py

+15-15
Original file line numberDiff line numberDiff line change
@@ -1221,21 +1221,21 @@ def assert_series_equal(
12211221

12221222
# This could be refactored to use the NDFrame.equals method
12231223
def assert_frame_equal(
1224-
left,
1225-
right,
1226-
check_dtype=True,
1227-
check_index_type="equiv",
1228-
check_column_type="equiv",
1229-
check_frame_type=True,
1230-
check_less_precise=False,
1231-
check_names=True,
1232-
by_blocks=False,
1233-
check_exact=False,
1234-
check_datetimelike_compat=False,
1235-
check_categorical=True,
1236-
check_like=False,
1237-
obj="DataFrame",
1238-
):
1224+
left: DataFrame,
1225+
right: DataFrame,
1226+
check_dtype: bool = True,
1227+
check_index_type: str = "equiv",
1228+
check_column_type: str = "equiv",
1229+
check_frame_type: bool = True,
1230+
check_less_precise: bool = False,
1231+
check_names: bool = True,
1232+
by_blocks: bool = False,
1233+
check_exact: bool = False,
1234+
check_datetimelike_compat: bool = False,
1235+
check_categorical: bool = True,
1236+
check_like: bool = False,
1237+
obj: str = "DataFrame",
1238+
) -> None:
12391239
"""
12401240
Check that left and right DataFrame are equal.
12411241

0 commit comments

Comments
 (0)