Testing make_multimodel_tableheaders (#88)

LSYS · Dec 15, 2023 · 8f504af · 8f504af
1 parent 9a79b54
commit 8f504af
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_mplot_dataframe_utils.py b/tests/test_mplot_dataframe_utils.py
@@ -207,4 +207,7 @@ def test_make_multimodel_tableheaders():
     )
     # Verify
     # assert_frame_equal(df_result, df_expected)
-    pd.testing.assert_frame_equal(df_result.iloc[:, :4], df_expected.iloc[:, :4])
+    assert_frame_equal(df_result.iloc[:, :4], df_expected.iloc[:, :4])
+    assert_series_equal(df_result["yticklabel"], df_expected["yticklabel"])
+    assert_series_equal(df_result["yticklabel2"], df_expected["yticklabel2"])
+