@@ -608,19 +608,22 @@ def test_pandas_iterable():
608608assert_array_equal (cm1 .colors ,cm2 .colors )
609609
610610
611- def test_colormap_reversing ():
611+ @pytest .mark .parametrize ('name' ,cm .cmap_d )
612+ def test_colormap_reversing (name ):
612613"""Check the generated _lut data of a colormap and corresponding
613614 reversed colormap if they are almost the same."""
614- for name in cm .cmap_d :
615- with warnings .catch_warnings (record = True )as w :
616- warnings .simplefilter ('always' )
617- cmap = plt .get_cmap (name )
618- assert len (w )== (1 if name in ('spectral' ,'spectral_r' )else 0 )
619- cmap_r = cmap .reversed ()
620- if not cmap_r ._isinit :
621- cmap ._init ()
622- cmap_r ._init ()
623- assert_array_almost_equal (cmap ._lut [:- 3 ],cmap_r ._lut [- 4 ::- 1 ])
615+ should_have_warning = {'spectral' ,'spectral_r' ,'Vega10' ,'Vega10_r' ,
616+ 'Vega20' ,'Vega20_r' ,'Vega20b' ,'Vega20b_r' ,
617+ 'Vega20c' ,'Vega20c_r' }
618+ with warnings .catch_warnings (record = True )as w :
619+ warnings .simplefilter ('always' )
620+ cmap = plt .get_cmap (name )
621+ assert len (w )== (1 if name in should_have_warning else 0 )
622+ cmap_r = cmap .reversed ()
623+ if not cmap_r ._isinit :
624+ cmap ._init ()
625+ cmap_r ._init ()
626+ assert_array_almost_equal (cmap ._lut [:- 3 ],cmap_r ._lut [- 4 ::- 1 ])
624627
625628
626629@cleanup