Skip to content

Commit 090957c

Browse files
ENH: Show column name in assert_frame_equal (#29218)
Co-authored-by: Joris Van den Bossche <[email protected]>
1 parent 40bff2f commit 090957c

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

pandas/_testing.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1287,7 +1287,7 @@ def assert_frame_equal(
12871287
>>> assert_frame_equal(df1, df2)
12881288
Traceback (most recent call last):
12891289
...
1290-
AssertionError: Attributes of DataFrame.iloc[:, 1] are different
1290+
AssertionError: Attributes of DataFrame.iloc[:, 1] (column name="b") are different
12911291
12921292
Attribute "dtype" are different
12931293
[left]: int64
@@ -1366,7 +1366,7 @@ def assert_frame_equal(
13661366
check_names=check_names,
13671367
check_datetimelike_compat=check_datetimelike_compat,
13681368
check_categorical=check_categorical,
1369-
obj=f"{obj}.iloc[:, {i}]",
1369+
obj=f'{obj}.iloc[:, {i}] (column name="{col}")',
13701370
)
13711371

13721372

pandas/tests/util/test_assert_frame_equal.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -181,9 +181,9 @@ def test_frame_equal_columns_mismatch(obj_fixture):
181181

182182

183183
def test_frame_equal_block_mismatch(by_blocks_fixture, obj_fixture):
184-
msg = """{obj}\\.iloc\\[:, 1\\] are different
184+
msg = """{obj}\\.iloc\\[:, 1\\] \\(column name="B"\\) are different
185185
186-
{obj}\\.iloc\\[:, 1\\] values are different \\(33\\.33333 %\\)
186+
{obj}\\.iloc\\[:, 1\\] \\(column name="B"\\) values are different \\(33\\.33333 %\\)
187187
\\[left\\]: \\[4, 5, 6\\]
188188
\\[right\\]: \\[4, 5, 7\\]""".format(
189189
obj=obj_fixture
@@ -202,18 +202,18 @@ def test_frame_equal_block_mismatch(by_blocks_fixture, obj_fixture):
202202
(
203203
DataFrame({"A": ["á", "à", "ä"], "E": ["é", "è", "ë"]}),
204204
DataFrame({"A": ["á", "à", "ä"], "E": ["é", "è", "e̊"]}),
205-
"""{obj}\\.iloc\\[:, 1\\] are different
205+
"""{obj}\\.iloc\\[:, 1\\] \\(column name="E"\\) are different
206206
207-
{obj}\\.iloc\\[:, 1\\] values are different \\(33\\.33333 %\\)
207+
{obj}\\.iloc\\[:, 1\\] \\(column name="E"\\) values are different \\(33\\.33333 %\\)
208208
\\[left\\]: \\[é, è, ë\\]
209209
\\[right\\]: \\[é, è, e̊\\]""",
210210
),
211211
(
212212
DataFrame({"A": ["á", "à", "ä"], "E": ["é", "è", "ë"]}),
213213
DataFrame({"A": ["a", "a", "a"], "E": ["e", "e", "e"]}),
214-
"""{obj}\\.iloc\\[:, 0\\] are different
214+
"""{obj}\\.iloc\\[:, 0\\] \\(column name="A"\\) are different
215215
216-
{obj}\\.iloc\\[:, 0\\] values are different \\(100\\.0 %\\)
216+
{obj}\\.iloc\\[:, 0\\] \\(column name="A"\\) values are different \\(100\\.0 %\\)
217217
\\[left\\]: \\[á, à, ä\\]
218218
\\[right\\]: \\[a, a, a\\]""",
219219
),

0 commit comments

Comments
 (0)