fix build failed

henryhu123 · henryhu123 · commit 0322cbd09605 · 2020-03-13T20:39:37.000-04:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -3998,12 +3998,12 @@ def generate_lines_with_colors_inputs():
     xy_mask = np.ma.masked_equal([1, 2, 3, 4, 5], 4)
     lines_nan = [{'base_xy': xy_nan,
                     'base_max': np.ones(5),
-                    'base_min': 0,
+                    'base_min': [0],
                     'linewidth': 5,
                     'colors': colors}]
     lines_mask = [{'base_xy': xy_mask,
                     'base_max': np.ones(5),
-                    'base_min': 0,
+                    'base_min': [0],
                     'linewidth': 5,
                     'colors': colors}]
 
@@ -4013,41 +4013,36 @@ def generate_lines_with_colors_inputs():
 @pytest.mark.parametrize('kwargs', generate_lines_with_colors_inputs())
 @check_figures_equal(extensions=["png"])
 def test_vlines_with_colors(fig_test, fig_ref, kwargs):
-    fig_test, ax0 = plt.subplots()
     kwargs['x'] = kwargs.pop('base_xy')
     kwargs['ymin'] = kwargs.pop('base_min')
     kwargs['ymax'] = kwargs.pop('base_max')
-    ax0.vlines(**kwargs)
+    fig_test.subplots().vlines(**kwargs)
 
     expect = {
     'x': [1, 2, 3, 5],
-    'ymin': 0,
-    'ymax': [1, 2, 3, 5],
+    'ymin': [0],
+    'ymax': np.ones(4),
     'colors': ['red', 'green', 'blue', 'orange'],
     'linewidth': 5}
-
-    fig_ref, ax1 = plt.subplots()
-    ax1.vlines(**expect)
+    fig_ref.subplots().vlines(**expect)
 
 
 @pytest.mark.parametrize('kwargs', generate_lines_with_colors_inputs())
 @check_figures_equal(extensions=["png"])
 def test_hlines_with_colors(fig_test, fig_ref, kwargs):
-    fig_test, ax0 = plt.subplots()
     kwargs['y'] = kwargs.pop('base_xy')
     kwargs['xmin'] = kwargs.pop('base_min')
     kwargs['xmax'] = kwargs.pop('base_max')
-    ax0.hlines(**kwargs)
+    fig_test.subplots().hlines(**kwargs)
 
     expect = {
     'y': [1, 2, 3, 5],
-    'xmin': 0,
-    'xmax': [1, 2, 3, 5],
+    'xmin': [0],
+    'xmax': np.ones(4),
     'colors': ['red', 'green', 'blue', 'orange'],
     'linewidth': 5}
 
-    fig_ref, ax1 = plt.subplots()
-    ax1.hlines(**expect)
+    fig_ref.subplots().hlines(**expect)
 
 
 @image_comparison(['step_linestyle', 'step_linestyle'], remove_text=True)
