@@ -191,70 +191,63 @@ def test_gridspec_make_colorbar():
191
191
plt .subplots_adjust (top = 0.95 , right = 0.95 , bottom = 0.2 , hspace = 0.25 )
192
192
193
193
194
- @image_comparison (baseline_images = ['colorbar_join_lsc' ,
195
- 'colorbar_join_lsc_frac' ,
196
- 'colorbar_join_listed' ,
197
- 'colorbar_join_listed_frac' , ],
194
+ @image_comparison (baseline_images = ['colorbar_join' ,
195
+ 'colorbar_join_frac' , ],
198
196
extensions = ['png' ], remove_text = True ,
199
197
savefig_kwarg = {'dpi' : 40 })
200
198
def test_join_colorbar ():
201
199
data = np .arange (1200 ).reshape (30 , 40 )
202
- levels = [0 , 200 , 400 , 600 , 800 , 1000 , 1200 ]
203
200
204
201
# Jet is a LinearSegmentedColormap
205
- cmap_lsc = plt .get_cmap ('jet' , 16 )
206
- cmap_lst = plt .get_cmap ('viridis' , 16 )
207
-
208
- # join returns the same type of cmap as self
209
- # Thus, this is a lsc
210
- cmap = cmap_lsc .join (cmap_lst )
211
-
212
- plt .figure ()
213
- plt .contourf (data , levels = levels , cmap = cmap )
214
- plt .colorbar (orientation = 'vertical' )
215
-
216
- # Use the 'frac_self' kwarg for the lsc cmap
217
- cmap = cmap_lsc .join (cmap_lst , frac_self = 0.7 )
218
-
219
- plt .figure ()
220
- plt .contourf (data , levels = levels , cmap = cmap )
221
- plt .colorbar (orientation = 'vertical' )
202
+ cmap1 = plt .get_cmap ('viridis' , 5 )
203
+ cmap2 = plt .get_cmap ('jet' , 5 )
222
204
223
205
# This should be a listed colormap.
224
- cmap = cmap_lst .join (cmap_lsc )
206
+ cmap = cmap1 .join (cmap2 )
225
207
226
208
plt .figure ()
227
- plt .contourf (data , levels = levels , cmap = cmap )
209
+ plt .pcolormesh (data , cmap = cmap )
228
210
plt .colorbar (orientation = 'vertical' )
229
211
230
212
# Use the 'frac_self' kwarg for the listed cmap
231
- cmap = cmap_lst .join (cmap_lsc , frac_self = 0.7 )
213
+ cmap = cmap1 .join (cmap2 , frac_self = 0.7 , N = 50 )
232
214
233
215
plt .figure ()
234
- plt .contourf (data , levels = levels , cmap = cmap )
216
+ plt .pcolormesh (data , cmap = cmap )
235
217
plt .colorbar (orientation = 'vertical' )
236
218
237
219
238
- @image_comparison (baseline_images = ['colorbar_truncate_lsc' ,
239
- 'colorbar_truncate_listed' , ],
220
+ @image_comparison (baseline_images = ['colorbar_truncate' ,
221
+ 'colorbar_trunc-getitem' ,
222
+ 'colorbar_trunc-getitem-int' ,
223
+ 'colorbar_trunc-getitem-int-1jN' , ],
240
224
extensions = ['png' ], remove_text = True ,
241
225
savefig_kwarg = {'dpi' : 40 })
242
226
def test_truncate_colorbar ():
243
227
data = np .arange (1200 ).reshape (30 , 40 )
244
- levels = [0 , 200 , 400 , 600 , 800 , 1000 , 1200 ]
245
228
246
- # jet is a LinearSegmentedColormap
247
- cmap = plt .get_cmap ('jet' , 16 ).truncate (0.2 , 0.7 )
229
+ cmap = plt .get_cmap ('viridis' , 32 ).truncate (0.2 , 0.7 )
230
+
231
+ plt .figure ()
232
+ plt .pcolormesh (data , cmap = cmap )
233
+ plt .colorbar (orientation = 'vertical' )
234
+
235
+ cmap = plt .get_cmap ('viridis' , 128 )[0.2 :- 0.3 :16 * 1j ]
236
+
237
+ plt .figure ()
238
+ plt .pcolormesh (data , cmap = cmap )
239
+ plt .colorbar (orientation = 'vertical' )
240
+
241
+ cmap = plt .get_cmap ('viridis' , 128 )[25 :90 ]
248
242
249
243
plt .figure ()
250
- plt .contourf (data , levels = levels , cmap = cmap )
244
+ plt .pcolormesh (data , cmap = cmap )
251
245
plt .colorbar (orientation = 'vertical' )
252
246
253
- # viridis is a ListedColormap
254
- cmap = plt .get_cmap ('viridis' , 16 ).truncate (0.2 , 0.7 )
247
+ cmap = plt .get_cmap ('viridis' , 128 )[25 :90 :16 * 1j ]
255
248
256
249
plt .figure ()
257
- plt .contourf (data , levels = levels , cmap = cmap )
250
+ plt .pcolormesh (data , cmap = cmap )
258
251
plt .colorbar (orientation = 'vertical' )
259
252
260
253
0 commit comments