def test_brain_colorbar(orientation, diverging, lims):
"""Test brain colorbar plotting."""
_, ax = plt.subplots()
clim = dict(kind='value')
if diverging:
clim['pos_lims'] = lims
else:
clim['lims'] = lims
plot_brain_colorbar(ax, clim, orientation=orientation)
if orientation == 'vertical':
have, empty = ax.get_yticklabels, ax.get_xticklabels
else:
have, empty = ax.get_xticklabels, ax.get_yticklabels
if diverging:
if lims[0] == 0:
ticks = list(-np.array(lims[1:][::-1])) + lims
else:
ticks = list(-np.array(lims[::-1])) + [0] + lims
else:
ticks = lims
plt.draw()
# old mpl always spans 0->1 for the actual ticks, so we need to
# look at the labels
assert_array_equal([float(h.get_text().replace('−', '-')) for h in have()],
ticks)
assert_array_equal(empty(), [])
plt.close('all')
def test_brain_colorbar(orientation, diverging, lims):
"""Test brain colorbar plotting."""
_, ax = plt.subplots()
clim = dict(kind='value')
if diverging:
clim['pos_lims'] = lims
else:
clim['lims'] = lims
cbar = plot_brain_colorbar(ax, clim, orientation=orientation)
ax = cbar.ax # in newer mpl this can be inset axes relative to the orig
if orientation == 'vertical':
have, empty = ax.get_yticklabels, ax.get_xticklabels
else:
have, empty = ax.get_xticklabels, ax.get_yticklabels
if diverging:
if lims[0] == 0:
ticks = list(-np.array(lims[1:][::-1])) + lims
else:
ticks = list(-np.array(lims[::-1])) + [0] + lims
else:
ticks = lims
ax.figure.canvas.draw_idle()
assert_array_equal([float(h.get_text().replace('−', '-')) for h in have()],
ticks)
assert_array_equal(empty(), [])
img = peak_stc.as_volume(src_fs, mri_resolution=False)
fig = plt.figure(figsize=(6, 2.5))
axes = [
plt.subplot2grid((4, 16), (0, 0), rowspan=4, colspan=14),
plt.subplot2grid((4, 16), (1, 15), rowspan=2, colspan=1)
]
plot_glass_brain(img,
title='t = 0.50s, Cue = A',
draw_cross=False,
annotate=True,
colorbar=False,
cmap=colormap,
threshold=lims[0],
vmax=lims[-1],
axes=axes[0])
plot_brain_colorbar(axes[1],
clim,
'magma_r',
label='Activation (NAI)',
bgcolor='white')
fig.subplots_adjust(left=0.05,
right=0.9,
bottom=0.01,
top=0.9,
wspace=0.5,
hspace=0.1)
fig.savefig(fname.figures + '/lcmv_A_500ms_lower.pdf', dpi=300)
plt.subplot2grid((6, 23), (0, 15), rowspan=6, colspan=3),
plt.subplot2grid((6, 23), (0, 18), rowspan=6, colspan=3),
plt.subplot2grid((6, 23), (1, 22), rowspan=4, colspan=1)
]
for ti, t in enumerate(ttp):
ab_diff.plot_topomap(times=t,
average=0.05,
vmin=-4,
vmax=4,
extrapolate='head',
colorbar=False,
axes=axes[ti],
show=False)
plot_brain_colorbar(axes[-1],
clim,
'RdBu_r',
label='Difference Cue B - Cue A',
bgcolor='darkblue')
fig.savefig(fname.figures + '/Diff_Topomaps.pdf', dpi=300)
###############################################################################
# 7) Plot ERPs for individual electrodes of interest
cis = within_subject_cis([a_erps, b_erps])
for electrode in [
'FCz', 'FC1', 'FC3', 'Cz', 'C1', 'C3', 'Pz', 'Oz', 'PO8', 'PO7'
]:
pick = ga_a_cue.ch_names.index(electrode)
fig, ax = plt.subplots(figsize=(8, 4))
plot_compare_evokeds(
