@@ -69,9 +69,8 @@ def test_plot_kwargs(self):
69
69
70
70
@pytest .mark .parametrize ("column, expected_axes_num" , [(None , 2 ), ("b" , 1 )])
71
71
@pytest .mark .parametrize ("label" , [None , "d" ])
72
- def test_groupby_hist_with_legend (self , column , expected_axes_num , label ):
73
- # GH 6279
74
- # Histogram can have a legend
72
+ def test_groupby_hist_frame_with_legend (self , column , expected_axes_num , label ):
73
+ # GH 6279 - Histogram can have a legend
75
74
expected_layout = (1 , expected_axes_num )
76
75
expected_labels = label or column or [["a" ], ["b" ]]
77
76
@@ -84,8 +83,7 @@ def test_groupby_hist_with_legend(self, column, expected_axes_num, label):
84
83
if label is not None :
85
84
kwargs ["label" ] = label
86
85
87
- ret = g .hist (** kwargs )
88
- for (_ , axes ) in ret .iteritems ():
86
+ for axes in g .hist (** kwargs ):
89
87
self ._check_axes_shape (
90
88
axes , axes_num = expected_axes_num , layout = expected_layout
91
89
)
@@ -96,8 +94,7 @@ def test_groupby_hist_with_legend(self, column, expected_axes_num, label):
96
94
"label, expected_label" , [(None , ["1" , "2" ]), ("d" , ["d" , "d" ])]
97
95
)
98
96
def test_groupby_hist_series_with_legend (self , label , expected_label ):
99
- # GH 6279
100
- # Histogram can have a legend
97
+ # GH 6279 - Histogram can have a legend
101
98
index = Index (15 * ["1" ] + 15 * ["2" ], name = "c" )
102
99
df = DataFrame (np .random .randn (30 , 2 ), index = index , columns = ["a" , "b" ])
103
100
g = df .groupby ("c" )
@@ -107,7 +104,6 @@ def test_groupby_hist_series_with_legend(self, label, expected_label):
107
104
if label is not None :
108
105
kwargs ["label" ] = label
109
106
110
- axes = g ["a" ].hist (** kwargs )
111
- for (_ , ax ) in axes .iteritems ():
107
+ for ax in g ["a" ].hist (** kwargs ):
112
108
self ._check_axes_shape (ax , axes_num = 1 , layout = (1 , 1 ))
113
109
self ._check_legend_labels (ax , expected_label )
0 commit comments