Skip to content

Commit a0ee437

Browse files
committed
Add more tests.
1 parent c98270e commit a0ee437

13 files changed

+28
-35
lines changed
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.

lib/matplotlib/tests/test_colorbar.py

+28-35
Original file line numberDiff line numberDiff line change
@@ -191,70 +191,63 @@ def test_gridspec_make_colorbar():
191191
plt.subplots_adjust(top=0.95, right=0.95, bottom=0.2, hspace=0.25)
192192

193193

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', ],
198196
extensions=['png'], remove_text=True,
199197
savefig_kwarg={'dpi': 40})
200198
def test_join_colorbar():
201199
data = np.arange(1200).reshape(30, 40)
202-
levels = [0, 200, 400, 600, 800, 1000, 1200]
203200

204201
# 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)
222204

223205
# This should be a listed colormap.
224-
cmap = cmap_lst.join(cmap_lsc)
206+
cmap = cmap1.join(cmap2)
225207

226208
plt.figure()
227-
plt.contourf(data, levels=levels, cmap=cmap)
209+
plt.pcolormesh(data, cmap=cmap)
228210
plt.colorbar(orientation='vertical')
229211

230212
# 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)
232214

233215
plt.figure()
234-
plt.contourf(data, levels=levels, cmap=cmap)
216+
plt.pcolormesh(data, cmap=cmap)
235217
plt.colorbar(orientation='vertical')
236218

237219

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', ],
240224
extensions=['png'], remove_text=True,
241225
savefig_kwarg={'dpi': 40})
242226
def test_truncate_colorbar():
243227
data = np.arange(1200).reshape(30, 40)
244-
levels = [0, 200, 400, 600, 800, 1000, 1200]
245228

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]
248242

249243
plt.figure()
250-
plt.contourf(data, levels=levels, cmap=cmap)
244+
plt.pcolormesh(data, cmap=cmap)
251245
plt.colorbar(orientation='vertical')
252246

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]
255248

256249
plt.figure()
257-
plt.contourf(data, levels=levels, cmap=cmap)
250+
plt.pcolormesh(data, cmap=cmap)
258251
plt.colorbar(orientation='vertical')
259252

260253

0 commit comments

Comments
 (0)