Improve tests as suggested in PR

Author: chahak13

lib/matplotlib/tests/test_legend.py

@@ -691,25 +691,19 @@ def test_legend_pathcollection_labelcolor_linecolor_iterable():
     ax.scatter(np.arange(10), np.arange(10)*1, label='#1', c=colors)
 
     leg = ax.legend(labelcolor='linecolor')
-    for text, color in zip(leg.get_texts(), ['k']):
-        assert mpl.colors.same_color(text.get_color(), color)
+    text, = leg.get_texts()
+    assert mpl.colors.same_color(text, 'black')
 
 
 def test_legend_pathcollection_labelcolor_linecolor_cmap():
     # test the labelcolor for labelcolor='linecolor' on PathCollection
     # with a colormap
     fig, ax = plt.subplots()
-    ax.scatter(
-        np.arange(10),
-        np.arange(10),
-        label='#1',
-        c=np.arange(10),
-        cmap="Reds"
-    )
+    ax.scatter(np.arange(10), np.arange(10), c=np.arange(10), label='#1')
 
     leg = ax.legend(labelcolor='linecolor')
-    for text, color in zip(leg.get_texts(), ['k']):
-        assert mpl.colors.same_color(text.get_color(), color)
+    text, = leg.get_texts()
+    assert mpl.colors.same_color(text, 'black')
 
 
 def test_legend_labelcolor_markeredgecolor():