def test_viz(): """Test viz.""" import matplotlib.pyplot as plt set_matplotlib_defaults(plt) events = mne.find_events(raw) picks = mne.pick_channels(raw.info['ch_names'], ['MEG 2443', 'MEG 2442', 'MEG 2441']) epochs = mne.Epochs(raw, events, picks=picks, baseline=(None, 0), reject=None, preload=True, event_id={'1': 1, '2': 2, '3': 3, '4': 4}) bad_epochs_idx = [0, 1, 3] n_epochs, n_channels, _ = epochs.get_data().shape bad_epochs = np.zeros(n_epochs, dtype=bool) bad_epochs[bad_epochs_idx] = True labels = np.zeros((n_epochs, n_channels)) reject_log = autoreject.RejectLog(bad_epochs, labels, epochs.ch_names) reject_log.plot_epochs(epochs) reject_log.plot() reject_log.plot(orientation='horizontal') pytest.raises(ValueError, reject_log.plot_epochs, epochs[:2]) pytest.raises(ValueError, reject_log.plot, 'down') plt.close('all') fig_in, ax = plt.subplots() fig_out = reject_log.plot(ax=ax) assert fig_in == fig_out
evoked = epochs.average() evoked_clean = epochs_clean.average() ############################################################################### # We will manually mark the bad channels just for plotting. evoked.info['bads'] = ['MEG 2443'] evoked_clean.info['bads'] = ['MEG 2443'] ############################################################################### # Let us plot the results. from autoreject.utils import set_matplotlib_defaults # noqa import matplotlib.pyplot as plt # noqa set_matplotlib_defaults(plt) fig, axes = plt.subplots(2, 1, figsize=(6, 6)) for ax in axes: ax.tick_params(axis='x', which='both', bottom='off', top='off') ax.tick_params(axis='y', which='both', left='off', right='off') ylim = dict(grad=(-170, 200)) evoked.pick_types(meg='grad', exclude=[]) evoked.plot(exclude=[], axes=axes[0], ylim=ylim, show=False) axes[0].set_title('Before RANSAC') evoked_clean.pick_types(meg='grad', exclude=[]) evoked_clean.plot(exclude=[], axes=axes[1], ylim=ylim) axes[1].set_title('After RANSAC') fig.tight_layout()