Skip to content

Commit ad99297

Browse files
committed
Be a little braver and change the original obj definition
1 parent 09a1ae5 commit ad99297

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pandas/_testing/asserters.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def assert_index_equal(
188188
check_order: bool = True,
189189
rtol: float = 1.0e-5,
190190
atol: float = 1.0e-8,
191-
obj: str = "Index",
191+
obj: str | None = None,
192192
) -> None:
193193
"""
194194
Check that left and right Index are equal.
@@ -217,7 +217,7 @@ def assert_index_equal(
217217
Relative tolerance. Only used when check_exact is False.
218218
atol : float, default 1e-8
219219
Absolute tolerance. Only used when check_exact is False.
220-
obj : str, default 'Index'
220+
obj : str, default 'Index' or 'MultiIndex'
221221
Specify object name being compared, internally used to show appropriate
222222
assertion message.
223223
@@ -235,6 +235,9 @@ def assert_index_equal(
235235
"""
236236
__tracebackhide__ = True
237237

238+
if obj is None:
239+
obj = "MultiIndex" if isinstance(left, MultiIndex) else "Index"
240+
238241
def _check_types(left, right, obj: str = "Index") -> None:
239242
if not exact:
240243
return
@@ -281,10 +284,9 @@ def _check_types(left, right, obj: str = "Index") -> None:
281284
# MultiIndex special comparison for little-friendly error messages
282285
if isinstance(left, MultiIndex):
283286
right = cast(MultiIndex, right)
284-
root_obj = "MultiIndex" if obj == "Index" else obj
285287

286288
for level in range(left.nlevels):
287-
lobj = f"{root_obj} level [{level}]"
289+
lobj = f"{obj} level [{level}]"
288290
try:
289291
# try comparison on levels/codes to avoid densifying MultiIndex
290292
assert_index_equal(

0 commit comments

Comments
 (0)