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

Commit66ec3a0

Browse files
committed
Add tests for legend with str args in scatter
Tests are added to check various conditions for the text color for alegend set with `linecolor`, `markeredgecolor`, `markerfacecolor` in ascatter plot.
1 parentb417a29 commit66ec3a0

File tree

1 file changed

+126
-0
lines changed

1 file changed

+126
-0
lines changed

‎lib/matplotlib/tests/test_legend.py

Lines changed: 126 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -671,6 +671,47 @@ def test_legend_labelcolor_linecolor():
671671
assertmpl.colors.same_color(text.get_color(),color)
672672

673673

674+
deftest_legend_pathcollection_labelcolor_linecolor():
675+
# test the labelcolor for labelcolor='linecolor' on PathCollection
676+
fig,ax=plt.subplots()
677+
ax.scatter(np.arange(10),np.arange(10)*1,label='#1',c='r')
678+
ax.scatter(np.arange(10),np.arange(10)*2,label='#2',c='g')
679+
ax.scatter(np.arange(10),np.arange(10)*3,label='#3',c='b')
680+
681+
leg=ax.legend(labelcolor='linecolor')
682+
fortext,colorinzip(leg.get_texts(), ['r','g','b']):
683+
assertmpl.colors.same_color(text.get_color(),color)
684+
685+
686+
deftest_legend_pathcollection_labelcolor_linecolor_iterable():
687+
# test the labelcolor for labelcolor='linecolor' on PathCollection
688+
# with iterable colors
689+
fig,ax=plt.subplots()
690+
colors=np.random.default_rng().choice(['r','g','b'],10)
691+
ax.scatter(np.arange(10),np.arange(10)*1,label='#1',c=colors)
692+
693+
leg=ax.legend(labelcolor='linecolor')
694+
fortext,colorinzip(leg.get_texts(), ['k']):
695+
assertmpl.colors.same_color(text.get_color(),color)
696+
697+
698+
deftest_legend_pathcollection_labelcolor_linecolor_cmap():
699+
# test the labelcolor for labelcolor='linecolor' on PathCollection
700+
# with a colormap
701+
fig,ax=plt.subplots()
702+
ax.scatter(
703+
np.arange(10),
704+
np.arange(10),
705+
label='#1',
706+
c=np.arange(10),
707+
cmap="Reds"
708+
)
709+
710+
leg=ax.legend(labelcolor='linecolor')
711+
fortext,colorinzip(leg.get_texts(), ['k']):
712+
assertmpl.colors.same_color(text.get_color(),color)
713+
714+
674715
deftest_legend_labelcolor_markeredgecolor():
675716
# test the labelcolor for labelcolor='markeredgecolor'
676717
fig,ax=plt.subplots()
@@ -683,6 +724,49 @@ def test_legend_labelcolor_markeredgecolor():
683724
assertmpl.colors.same_color(text.get_color(),color)
684725

685726

727+
deftest_legend_pathcollection_labelcolor_markeredgecolor():
728+
# test the labelcolor for labelcolor='markeredgecolor' on PathCollection
729+
fig,ax=plt.subplots()
730+
ax.scatter(np.arange(10),np.arange(10)*1,label='#1',edgecolor='r')
731+
ax.scatter(np.arange(10),np.arange(10)*2,label='#2',edgecolor='g')
732+
ax.scatter(np.arange(10),np.arange(10)*3,label='#3',edgecolor='b')
733+
734+
leg=ax.legend(labelcolor='markeredgecolor')
735+
fortext,colorinzip(leg.get_texts(), ['r','g','b']):
736+
assertmpl.colors.same_color(text.get_color(),color)
737+
738+
739+
deftest_legend_pathcollection_labelcolor_markeredgecolor_iterable():
740+
# test the labelcolor for labelcolor='markeredgecolor' on PathCollection
741+
# with iterable colors
742+
fig,ax=plt.subplots()
743+
colors=np.random.default_rng().choice(['r','g','b'],10)
744+
ax.scatter(np.arange(10),np.arange(10)*1,label='#1',edgecolor=colors)
745+
746+
leg=ax.legend(labelcolor='markeredgecolor')
747+
fortext,colorinzip(leg.get_texts(), ['k']):
748+
assertmpl.colors.same_color(text.get_color(),color)
749+
750+
751+
deftest_legend_pathcollection_labelcolor_markeredgecolor_cmap():
752+
# test the labelcolor for labelcolor='markeredgecolor' on PathCollection
753+
# with a colormap
754+
fig,ax=plt.subplots()
755+
edgecolors=mpl.cm.viridis(np.random.rand(10))
756+
ax.scatter(
757+
np.arange(10),
758+
np.arange(10),
759+
label='#1',
760+
c=np.arange(10),
761+
edgecolor=edgecolors,
762+
cmap="Reds"
763+
)
764+
765+
leg=ax.legend(labelcolor='markeredgecolor')
766+
fortext,colorinzip(leg.get_texts(), ['k']):
767+
assertmpl.colors.same_color(text.get_color(),color)
768+
769+
686770
deftest_legend_labelcolor_markerfacecolor():
687771
# test the labelcolor for labelcolor='markerfacecolor'
688772
fig,ax=plt.subplots()
@@ -695,6 +779,48 @@ def test_legend_labelcolor_markerfacecolor():
695779
assertmpl.colors.same_color(text.get_color(),color)
696780

697781

782+
deftest_legend_pathcollection_labelcolor_markerfacecolor():
783+
# test the labelcolor for labelcolor='markerfacecolor' on PathCollection
784+
fig,ax=plt.subplots()
785+
ax.scatter(np.arange(10),np.arange(10)*1,label='#1',facecolor='r')
786+
ax.scatter(np.arange(10),np.arange(10)*2,label='#2',facecolor='g')
787+
ax.scatter(np.arange(10),np.arange(10)*3,label='#3',facecolor='b')
788+
789+
leg=ax.legend(labelcolor='markerfacecolor')
790+
fortext,colorinzip(leg.get_texts(), ['r','g','b']):
791+
assertmpl.colors.same_color(text.get_color(),color)
792+
793+
794+
deftest_legend_pathcollection_labelcolor_markerfacecolor_iterable():
795+
# test the labelcolor for labelcolor='markerfacecolor' on PathCollection
796+
# with iterable colors
797+
fig,ax=plt.subplots()
798+
colors=np.random.default_rng().choice(['r','g','b'],10)
799+
ax.scatter(np.arange(10),np.arange(10)*1,label='#1',facecolor=colors)
800+
801+
leg=ax.legend(labelcolor='markerfacecolor')
802+
fortext,colorinzip(leg.get_texts(), ['k']):
803+
assertmpl.colors.same_color(text.get_color(),color)
804+
805+
806+
deftest_legend_pathcollection_labelcolor_markfacecolor_cmap():
807+
# test the labelcolor for labelcolor='markerfacecolor' on PathCollection
808+
# with colormaps
809+
fig,ax=plt.subplots()
810+
facecolors=mpl.cm.viridis(np.random.rand(10))
811+
ax.scatter(
812+
np.arange(10),
813+
np.arange(10),
814+
label='#1',
815+
c=np.arange(10),
816+
facecolor=facecolors
817+
)
818+
819+
leg=ax.legend(labelcolor='markerfacecolor')
820+
fortext,colorinzip(leg.get_texts(), ['k']):
821+
assertmpl.colors.same_color(text.get_color(),color)
822+
823+
698824
@pytest.mark.parametrize('color', ('red','none', (.5,.5,.5)))
699825
deftest_legend_labelcolor_rcparam_single(color):
700826
# test the rcParams legend.labelcolor for a single color

0 commit comments

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