Skip to content

Commit 9ffdf03

Browse files
jzwickJustine Wezenaar
andauthored
TST: Add test for assert_frame_equal when nested DataFrame contains NA (#62628)
Co-authored-by: Justine Wezenaar <justinedzed@gmail.com>
1 parent 54e68a0 commit 9ffdf03

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

pandas/tests/util/test_assert_frame_equal.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import numpy as np
12
import pytest
23

34
from pandas.errors import Pandas4Warning
@@ -413,3 +414,12 @@ def test_datetimelike_compat_deprecated():
413414
tm.assert_series_equal(df["a"], df["a"], check_datetimelike_compat=True)
414415
with tm.assert_produces_warning(Pandas4Warning, match=msg):
415416
tm.assert_series_equal(df["a"], df["a"], check_datetimelike_compat=False)
417+
418+
419+
@pytest.mark.parametrize("na_value", [pd.NA, np.nan, None])
420+
def test_assert_frame_equal_nested_df_na(na_value):
421+
# GH#43022
422+
inner = DataFrame({"a": [1, na_value]})
423+
df1 = DataFrame({"df": [inner]})
424+
df2 = DataFrame({"df": [inner]})
425+
tm.assert_frame_equal(df1, df2)

0 commit comments

Comments
 (0)