Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitdabaf62

Browse files
committed
add tests for various corner cases of legends withlabelcolor set tolinecolor, e.g. ensuring that histograms withstep histtype do not end up with an empty label
1 parent2b75c9e commitdabaf62

File tree

1 file changed

+238
-0
lines changed

1 file changed

+238
-0
lines changed

‎lib/matplotlib/tests/test_legend.py

Lines changed: 238 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1068,6 +1068,244 @@ def test_legend_labelcolor_rcparam_markerfacecolor_short():
10681068
assertmpl.colors.same_color(text.get_color(),color)
10691069

10701070

1071+
deftest_legend_labelcolor_linecolor_histograms():
1072+
x=np.arange(10)
1073+
1074+
# testing c kwarg for bar, step, and stepfilled histograms
1075+
fig,ax=plt.subplots()
1076+
_,_,h=ax.hist(x,histtype='bar',color='r',
1077+
label="red bar hist with a red label")
1078+
leg=ax.legend(loc=1,labelcolor='linecolor')
1079+
tc=leg.texts[0].get_color()
1080+
assertmpl.colors.same_color(tc,'r')
1081+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1082+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1083+
1084+
fig,ax=plt.subplots()
1085+
_,_,h=ax.hist(x,histtype='step',color='g',
1086+
label="green step hist with a green label")
1087+
leg=ax.legend(loc=1,labelcolor='linecolor')
1088+
tc=leg.texts[0].get_color()
1089+
assertmpl.colors.same_color(tc,'g')
1090+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_edgecolor())
1091+
assertmpl.colors.same_color(tc,h[0].get_edgecolor())
1092+
1093+
fig,ax=plt.subplots()
1094+
_,_,h=ax.hist(x,histtype='stepfilled',color='b',
1095+
label="blue stepfilled hist with a blue label")
1096+
leg=ax.legend(loc=1,labelcolor='linecolor')
1097+
tc=leg.texts[0].get_color()
1098+
assertmpl.colors.same_color(tc,'b')
1099+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1100+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1101+
1102+
# testing c, fc, and ec combinations for bar histograms
1103+
fig,ax=plt.subplots()
1104+
_,_,h=ax.hist(x,histtype='bar',color='r',ec='b',
1105+
label="red bar hist with blue edges and a red label")
1106+
leg=ax.legend(loc=1,labelcolor='linecolor')
1107+
tc=leg.texts[0].get_color()
1108+
assertmpl.colors.same_color(tc,'r')
1109+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1110+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1111+
1112+
fig,ax=plt.subplots()
1113+
_,_,h=ax.hist(x,histtype='bar',fc='r',ec='b',
1114+
label="red bar hist with blue edges and a red label")
1115+
leg=ax.legend(loc=1,labelcolor='linecolor')
1116+
tc=leg.texts[0].get_color()
1117+
assertmpl.colors.same_color(tc,'r')
1118+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1119+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1120+
1121+
fig,ax=plt.subplots()
1122+
_,_,h=ax.hist(x,histtype='bar',fc='none',ec='b',
1123+
label="unfilled blue bar hist with a blue label")
1124+
leg=ax.legend(loc=1,labelcolor='linecolor')
1125+
tc=leg.texts[0].get_color()
1126+
assertmpl.colors.same_color(tc,'b')
1127+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_edgecolor())
1128+
assertmpl.colors.same_color(tc,h[0].get_edgecolor())
1129+
1130+
# testing c, and ec combinations for step histograms
1131+
fig,ax=plt.subplots()
1132+
_,_,h=ax.hist(x,histtype='step',color='r',ec='b',
1133+
label="blue step hist with a blue label")
1134+
leg=ax.legend(loc=1,labelcolor='linecolor')
1135+
tc=leg.texts[0].get_color()
1136+
assertmpl.colors.same_color(tc,'b')
1137+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_edgecolor())
1138+
assertmpl.colors.same_color(tc,h[0].get_edgecolor())
1139+
1140+
fig,ax=plt.subplots()
1141+
_,_,h=ax.hist(x,histtype='step',ec='b',
1142+
label="blue step hist with a blue label")
1143+
leg=ax.legend(loc=1,labelcolor='linecolor')
1144+
tc=leg.texts[0].get_color()
1145+
assertmpl.colors.same_color(tc,'b')
1146+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_edgecolor())
1147+
assertmpl.colors.same_color(tc,h[0].get_edgecolor())
1148+
1149+
# testing c, fc, and ec combinations for stepfilled histograms
1150+
fig,ax=plt.subplots()
1151+
_,_,h=ax.hist(x,histtype='stepfilled',color='r',ec='b',
1152+
label="red stepfilled hist, blue edges, red label")
1153+
leg=ax.legend(loc=1,labelcolor='linecolor')
1154+
tc=leg.texts[0].get_color()
1155+
assertmpl.colors.same_color(tc,'r')
1156+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1157+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1158+
1159+
fig,ax=plt.subplots()
1160+
_,_,h=ax.hist(x,histtype='stepfilled',fc='r',ec='b',
1161+
label="red stepfilled hist, blue edges, red label")
1162+
leg=ax.legend(loc=1,labelcolor='linecolor')
1163+
tc=leg.texts[0].get_color()
1164+
assertmpl.colors.same_color(tc,'r')
1165+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1166+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1167+
1168+
fig,ax=plt.subplots()
1169+
_,_,h=ax.hist(x,histtype='stepfilled',fc='none',ec='b',
1170+
label="unfilled blue stepfilled hist, blue label")
1171+
leg=ax.legend(loc=1,labelcolor='linecolor')
1172+
tc=leg.texts[0].get_color()
1173+
assertmpl.colors.same_color(tc,'b')
1174+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_edgecolor())
1175+
assertmpl.colors.same_color(tc,h[0].get_edgecolor())
1176+
1177+
fig,ax=plt.subplots()
1178+
_,_,h=ax.hist(x,histtype='stepfilled',fc='r',ec='none',
1179+
label="edgeless red stepfilled hist with a red label")
1180+
leg=ax.legend(loc=1,labelcolor='linecolor')
1181+
tc=leg.texts[0].get_color()
1182+
assertmpl.colors.same_color(tc,'r')
1183+
assertmpl.colors.same_color(tc,leg.get_patches()[0].get_facecolor())
1184+
assertmpl.colors.same_color(tc,h[0].get_facecolor())
1185+
plt.close('all')
1186+
1187+
1188+
deftest_legend_labelcolor_linecolor_plot():
1189+
x=np.arange(5)
1190+
1191+
# testing line plot
1192+
fig,ax=plt.subplots()
1193+
p=ax.plot(x,c='r',label="red line with a red label")
1194+
leg=ax.legend(loc=1,labelcolor='linecolor')
1195+
tc=leg.texts[0].get_color()
1196+
assertmpl.colors.same_color(tc,'r')
1197+
assertmpl.colors.same_color(tc,leg.get_lines()[0].get_color())
1198+
assertmpl.colors.same_color(tc,p[0].get_color())
1199+
1200+
# testing c, fc, and ec combinations for maker plots
1201+
fig,ax=plt.subplots()
1202+
p=ax.plot(x,'o',c='r',label="red circles with a red label")
1203+
leg=ax.legend(loc=1,labelcolor='linecolor')
1204+
tc=leg.texts[0].get_color()
1205+
assertmpl.colors.same_color(tc,'r')
1206+
assertmpl.colors.same_color(tc,leg.get_lines()[0].get_color())
1207+
assertmpl.colors.same_color(tc,p[0].get_color())
1208+
1209+
fig,ax=plt.subplots()
1210+
p=ax.plot(x,'o',c='r',mec='b',label="red circles, blue edges, red label")
1211+
leg=ax.legend(loc=1,labelcolor='linecolor')
1212+
tc=leg.texts[0].get_color()
1213+
assertmpl.colors.same_color(tc,'r')
1214+
assertmpl.colors.same_color(tc,leg.get_lines()[0].get_color())
1215+
assertmpl.colors.same_color(tc,p[0].get_color())
1216+
1217+
fig,ax=plt.subplots()
1218+
p=ax.plot(x,'o',mfc='r',mec='b',label="red circles, blue edges, red label")
1219+
leg=ax.legend(loc=1,labelcolor='linecolor')
1220+
tc=leg.texts[0].get_color()
1221+
assertmpl.colors.same_color(tc,'r')
1222+
assertmpl.colors.same_color(tc,leg.get_lines()[0].get_markerfacecolor())
1223+
assertmpl.colors.same_color(tc,p[0].get_markerfacecolor())
1224+
1225+
# 'none' cases
1226+
fig,ax=plt.subplots()
1227+
p=ax.plot(x,'o',mfc='none',mec='b',label="blue unfilled circles, blue label")
1228+
leg=ax.legend(loc=1,labelcolor='linecolor')
1229+
tc=leg.texts[0].get_color()
1230+
assertmpl.colors.same_color(tc,'b')
1231+
assertmpl.colors.same_color(tc,leg.get_lines()[0].get_markeredgecolor())
1232+
assertmpl.colors.same_color(tc,p[0].get_markeredgecolor())
1233+
1234+
fig,ax=plt.subplots()
1235+
p=ax.plot(x,'o',mfc='r',mec='none',label="red edgeless circles, red label")
1236+
leg=ax.legend(loc=1,labelcolor='linecolor')
1237+
tc=leg.texts[0].get_color()
1238+
assertmpl.colors.same_color(tc,'r')
1239+
assertmpl.colors.same_color(tc,leg.get_lines()[0].get_markerfacecolor())
1240+
assertmpl.colors.same_color(tc,p[0].get_markerfacecolor())
1241+
1242+
fig,ax=plt.subplots()
1243+
p=ax.plot(x,'o',c='none',label="invisible circles with invisible label")
1244+
leg=ax.legend(loc=1,labelcolor='linecolor')
1245+
tc=leg.texts[0].get_color()
1246+
asserttc=='none'
1247+
asserttc==leg.get_lines()[0].get_markerfacecolor()
1248+
asserttc==leg.get_lines()[0].get_markeredgecolor()
1249+
asserttc==leg.get_lines()[0].get_color()
1250+
asserttc==p[0].get_markerfacecolor()
1251+
asserttc==p[0].get_markeredgecolor()
1252+
asserttc==p[0].get_color()
1253+
plt.close('all')
1254+
1255+
1256+
deftest_legend_labelcolor_linecolor_scatter():
1257+
x=np.arange(5)
1258+
1259+
# testing c, fc, and ec combinations for scatter plots
1260+
fig,ax=plt.subplots()
1261+
p=ax.scatter(x,x,c='r',label="red circles with a red label")
1262+
leg=ax.legend(loc=1,labelcolor='linecolor')
1263+
tc=leg.texts[0].get_color()
1264+
assertmpl.colors.same_color(tc,'r')
1265+
assertmpl.colors.same_color(tc,leg.legend_handles[0].get_facecolor())
1266+
assertmpl.colors.same_color(tc,p.get_facecolor())
1267+
1268+
fig,ax=plt.subplots()
1269+
p=ax.scatter(x,x,c='r',ec='b',label="red circles, blue edges, red label")
1270+
leg=ax.legend(loc=1,labelcolor='linecolor')
1271+
tc=leg.texts[0].get_color()
1272+
assertmpl.colors.same_color(tc,'r')
1273+
assertmpl.colors.same_color(tc,leg.legend_handles[0].get_facecolor())
1274+
assertmpl.colors.same_color(tc,p.get_facecolor())
1275+
1276+
fig,ax=plt.subplots()
1277+
p=ax.scatter(x,x,fc='r',ec='b',label="red circles, blue edges, red label")
1278+
leg=ax.legend(loc=1,labelcolor='linecolor')
1279+
tc=leg.texts[0].get_color()
1280+
assertmpl.colors.same_color(tc,'r')
1281+
assertmpl.colors.same_color(tc,leg.legend_handles[0].get_facecolor())
1282+
assertmpl.colors.same_color(tc,p.get_facecolor())
1283+
1284+
# 'none' cases
1285+
fig,ax=plt.subplots()
1286+
p=ax.scatter(x,x,fc='none',ec='b',label="blue unfilled circles, blue label")
1287+
leg=ax.legend(loc=1,labelcolor='linecolor')
1288+
tc=leg.texts[0].get_color()
1289+
assertmpl.colors.same_color(tc,'b')
1290+
assertmpl.colors.same_color(tc,leg.legend_handles[0].get_edgecolor())
1291+
assertmpl.colors.same_color(tc,p.get_edgecolor())
1292+
1293+
fig,ax=plt.subplots()
1294+
p=ax.scatter(x,x,fc='r',ec='none',label="red edgeless circles, red label")
1295+
leg=ax.legend(loc=1,labelcolor='linecolor')
1296+
tc=leg.texts[0].get_color()
1297+
assertmpl.colors.same_color(tc,'r')
1298+
assertmpl.colors.same_color(tc,leg.legend_handles[0].get_facecolor())
1299+
assertmpl.colors.same_color(tc,p.get_facecolor())
1300+
1301+
fig,ax=plt.subplots()
1302+
p=ax.scatter(x,x,c='none',label="invisible circles with invisible label")
1303+
leg=ax.legend(loc=1,labelcolor='linecolor')
1304+
tc=leg.texts[0].get_color()
1305+
asserttc=='none'
1306+
plt.close('all')
1307+
1308+
10711309
@pytest.mark.filterwarnings("ignore:No artists with labels found to put in legend")
10721310
deftest_get_set_draggable():
10731311
legend=plt.legend()

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp