assert_frame_equal for set

pandas/_libs/testing.pyx

@@ -95,6 +95,10 @@ cpdef assert_almost_equal(a, b,
     if robj is None:
         robj = b
 
+    if isinstance(a, set) or isinstance(b, set):
+        assert a == b, f"{a} != {b}"
+        return True
+
     if isinstance(a, dict) or isinstance(b, dict):
         return assert_dict_equal(a, b)
 

pandas/tests/util/test_assert_almost_equal.py

@@ -202,6 +202,16 @@ def test_assert_almost_equal_edge_case_ndarrays(left_dtype, right_dtype):
     )
 
 
+def test_assert_almost_equal_sets():
+    _assert_almost_equal_both({1, 2, 3}, {1, 2, 3})
+
+
+def test_assert_almost_not_equal_sets():
+    msg = r"{1, 2, 3} != {1, 2, 4}"
+    with pytest.raises(AssertionError, match=msg):
+        _assert_almost_equal_both({1, 2, 3}, {1, 2, 4})
+
+
 def test_assert_almost_equal_dicts():
     _assert_almost_equal_both({"a": 1, "b": 2}, {"a": 1, "b": 2})
 

pandas/tests/util/test_assert_frame_equal.py

@@ -362,3 +362,18 @@ def test_assert_frame_equal_ts_column():
     msg = r'DataFrame.iloc\[:, 0\] \(column name="a"\) values are different'
     with pytest.raises(AssertionError, match=msg):
         tm.assert_frame_equal(df1, df2)
+
+
+def test_assert_frame_equal_set():
+    df1 = DataFrame({"set_column": [{1, 2, 3}, {4, 5, 6}]})
+    df2 = DataFrame({"set_column": [{1, 2, 3}, {4, 5, 6}]})
+    tm.assert_frame_equal(df1, df2)
+
+
+def test_assert_frame_equal_set_mismatch():
+    df1 = DataFrame({"set_column": [{1, 2, 3}, {4, 5, 6}]})
+    df2 = DataFrame({"set_column": [{1, 2, 3}, {4, 5, 7}]})
+
+    msg = r'DataFrame.iloc\[:, 0\] \(column name="set_column"\) values are different'
+    with pytest.raises(AssertionError, match=msg):
+        tm.assert_frame_equal(df1, df2)