Fixed issue where expected_total_height returns as a df instead of a series

eicchen · eicchen · commit 16b6792ebca9 · 2025-05-02T21:48:24.000Z
diff --git a/pandas/tests/plotting/test_misc.py b/pandas/tests/plotting/test_misc.py
@@ -729,7 +729,7 @@ def _df_bar_subplot_checker(df_bar_data, df_bar_df, subplot_data_df, subplot_col
     ]
 
     if len(subplot_columns) == 1:
-        expected_total_height = df_bar_df.loc[:, subplot_columns]
+        expected_total_height = df_bar_df.loc[:, subplot_columns[0]]
     else:
         expected_total_height = df_bar_df.loc[:, subplot_columns].sum(axis=1)
 
@@ -747,7 +747,6 @@ def _df_bar_subplot_checker(df_bar_data, df_bar_df, subplot_data_df, subplot_col
             tm.assert_series_equal(
                 height_iter, expected_total_height, check_names=False, check_dtype=False
             )
-
         else:
             # Checks each preceding bar ends where the next one starts
             next_start_coord = subplot_sliced_by_source[i + 1]["y_coord"]

Original file line number	Diff line number	Diff line change
`@@ -729,7 +729,7 @@ def _df_bar_subplot_checker(df_bar_data, df_bar_df, subplot_data_df, subplot_col`
`729`	`729`	`]`
`730`	`730`
`731`	`731`	`if len(subplot_columns) == 1:`
`732`		`- expected_total_height = df_bar_df.loc[:, subplot_columns]`
	`732`	`+ expected_total_height = df_bar_df.loc[:, subplot_columns[0]]`
`733`	`733`	`else:`
`734`	`734`	`expected_total_height = df_bar_df.loc[:, subplot_columns].sum(axis=1)`
`735`	`735`
`@@ -747,7 +747,6 @@ def _df_bar_subplot_checker(df_bar_data, df_bar_df, subplot_data_df, subplot_col`
`747`	`747`	`tm.assert_series_equal(`
`748`	`748`	`height_iter, expected_total_height, check_names=False, check_dtype=False`
`749`	`749`	`)`
`750`		`-`
`751`	`750`	`else:`
`752`	`751`	`# Checks each preceding bar ends where the next one starts`
`753`	`752`	`next_start_coord = subplot_sliced_by_source[i + 1]["y_coord"]`