extend 'add test for the new function 'validate_grid_linestyle'

afvincent · afvincent · commit 5be29b5015c1 · 2017-02-07T16:46:27.000+01:00
diff --git a/lib/matplotlib/tests/test_rcparams.py b/lib/matplotlib/tests/test_rcparams.py
@@ -29,7 +29,8 @@
                                 validate_nseq_float,
                                 validate_cycler,
                                 validate_hatch,
-                                validate_hist_bins)
+                                validate_hist_bins,
+                                validate_grid_linestyle)
 
 
 mpl.rc('text', usetex=False)
@@ -333,6 +334,21 @@ def generate_validator_testcases(valid):
                      ),
          'fail': (('aardvark', ValueError),
                   )
+         },
+         {'validator': validate_grid_linestyle,
+         'success': (('--', '--'),
+                     ('dotted', 'dotted'),
+                     ('aardvark', 'aardvark'),
+                     (['1.23', '4.56'], (None, [1.23, 4.56])),
+                     ([1.23, 456], (None, [1.23, 456.0])),
+                     ([1, 2, 3, 4], (None, [1.0, 2.0, 3.0, 4.0])),
+                     ),
+         'fail': (((None, [1, 2]), ValueError),
+                  ((0, [1, 2]), ValueError),
+                  ((-1, [1, 2]), ValueError),
+                  ([1, 2, 3], ValueError),
+                  (1.23, ValueError)
+                  )
          }
     )
 
