From 8f504af850c041ee62a58df5770fa93c43fcce13 Mon Sep 17 00:00:00 2001 From: LSYS Date: Fri, 15 Dec 2023 11:42:14 +0800 Subject: [PATCH] Testing make_multimodel_tableheaders (#88) --- tests/test_mplot_dataframe_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_mplot_dataframe_utils.py b/tests/test_mplot_dataframe_utils.py index 3a8052a..666799f 100644 --- 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"]) + \ No newline at end of file