New check for the no len case. Now artists are stored

borgesaugusto · borgesaugusto · commit 27413fa5aeef · 2023-10-06T15:31:00.000+02:00
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -1360,7 +1360,18 @@ def test_loc_validation_string_value():
 
 
 def test_legend_handle_label_mismatch():
-    pl1 = plt.plot(range(10))
-    pl2 = plt.plot(range(10))
+    pl1, = plt.plot(range(10))
+    pl2, = plt.plot(range(10))
     with pytest.warns(UserWarning, match="number of handles and labels"):
-        plt.legend(handles=[pl1, pl2], labels=["pl1", "pl2", "pl3"])
+        legend = plt.legend(handles=[pl1, pl2], labels=["pl1", "pl2", "pl3"])
+        assert len(legend.legend_handles) == 2
+        assert len(legend.get_texts()) == 2
+
+
+def test_legend_handle_label_mismatch_no_len():
+    pl1, = plt.plot(range(10))
+    pl2, = plt.plot(range(10))
+    legend = plt.legend(handles=iter([pl1, pl2]),
+                        labels=iter(["pl1", "pl2", "pl3"]))
+    assert len(legend.legend_handles) == 2
+    assert len(legend.get_texts()) == 2
