Make tests more readable

yozhikoff · yozhikoff · commit dee18de01d0a · 2020-12-23T22:56:27.000+03:00
diff --git a/doc/users/next_whats_new/multiple_labels_for_Axes_plot.rst b/doc/users/next_whats_new/multiple_labels_for_Axes_plot.rst
@@ -9,9 +9,9 @@ When plotting multiple datasets by passing 2D data as *y* value to `~.Axes.plot`
     
     x = [1, 2, 3]
 
-    y = [[1, 9],
-         [2, 8],
-         [4, 6]]
+    y = [[1, 2],
+         [2, 5],
+         [4, 9]]
 
     plt.plot(x, y, label=['low', 'high'])
     plt.legend()
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -676,53 +676,49 @@ def test_no_warn_big_data_when_loc_specified():
 def test_plot_multiple_input_multiple_label():
     # test ax.plot() with multidimensional input
     # and multiple labels
-    x = [1, 2, 5]
-    y = [[2, 4, 3], [4, 7, 1], [3, 9, 2]]
-
-    label_arrays = [['one', 'two', 'three'],
-                    ('one', 'two', 'three'),
-                    np.array(['one', 'two', 'three'])]
-
+    x = [1, 2, 3]
+    y = [[1, 2],
+         [2, 5],
+         [4, 9]]
+    label_arrays = [['low', 'high'],
+                    ('low', 'high'),
+                    np.array(['low', 'high'])]
     for label in label_arrays:
         fig, ax = plt.subplots()
         ax.plot(x, y, label=label)
         leg = ax.legend()
-
         legend_texts = [entry.get_text() for entry in leg.get_texts()]
-        assert legend_texts == ['one', 'two', 'three']
+        assert legend_texts == ['low', 'high']
 
 
 def test_plot_multiple_input_single_label():
     # test ax.plot() with multidimensional input
     # and single label
-    x = [1, 2, 5]
-    y = [[2, 4, 3], [4, 7, 1], [3, 9, 2]]
+    x = [1, 2, 3]
+    y = [[1, 2],
+         [2, 5],
+         [4, 9]]
     labels = ['one', 1, int]
-
     for label in labels:
         fig, ax = plt.subplots()
         ax.plot(x, y, label=label)
         leg = ax.legend()
-
         legend_texts = [entry.get_text() for entry in leg.get_texts()]
-        assert legend_texts == [str(label)] * 3
+        assert legend_texts == [str(label)] * 2
 
 
 def test_plot_single_input_multiple_label():
     # test ax.plot() with 1D array like input
     # and iterable label
-    x = [1, 2, 5]
-    y = [2, 4, 3]
-
-    label_arrays = [['one', 'two', 'three'],
-                    ('one', 'two', 'three'),
-                    np.array(['one', 'two', 'three'])]
-
+    x = [1, 2, 3]
+    y = [2, 5, 6]
+    label_arrays = [['low', 'high'],
+                    ('low', 'high'),
+                    np.array(['low', 'high'])]
     for label in label_arrays:
         fig, ax = plt.subplots()
         ax.plot(x, y, label=label)
         leg = ax.legend()
-
         assert len(leg.get_texts()) == 1
         assert leg.get_texts()[0].get_text() == str(label)
 
@@ -731,8 +727,10 @@ def test_plot_multiple_label_incorrect_length_exception():
     # check that excepton is raised if multiple labels
     # are given, but number of on labels != number of lines
     with pytest.raises(ValueError):
-        x = [1, 2, 5]
-        y = [[2, 4, 3], [4, 7, 1], [3, 9, 2]]
-        label = ['one', 'two']
+        x = [1, 2, 3]
+        y = [[1, 2],
+             [2, 5],
+             [4, 9]]
+        label = ['high', 'low', 'medium']
         fig, ax = plt.subplots()
         ax.plot(x, y, label=label)
