def _check_axes(self, kwargs): '''Handle axes checking for Topo.''' # handle multiple axes self.multi_axes = self.values.ndim > 1 and self.values.shape[1] > 1 if self.multi_axes: self.n_topos = self.values.shape[1] else: self.n_topos = 1 has_axis = 'axes' in kwargs.keys() and kwargs['axes'] is not None if has_axis: # axis was passed, check if valid axes = kwargs['axes'] if self.multi_axes: # multiple topos and axes were passed, check if axes correct from mne.viz.utils import _validate_if_list_of_axes _validate_if_list_of_axes(axes, obligatory_len=self.n_topos) self.axes = axes else: # one topo and axes were passed, should check axes if self._squeezed and isinstance(axes, list): axes = axes[0] kwargs['axes'] = axes self.axes = [axes] else: # no axes passed, create figure and axes if self.multi_axes: # create a row of topographies n_per_topo = 2.5 figsize = (n_per_topo * self.n_topos, n_per_topo) fig, axes = plt.subplots(ncols=self.n_topos, figsize=figsize) self.axes = axes.tolist() else: fig, axes = plt.subplots() self.axes = [axes]
def test_validate_if_list_of_axes(): import matplotlib.pyplot as plt fig, ax = plt.subplots(2, 2) assert_raises(ValueError, _validate_if_list_of_axes, ax) ax_flat = ax.ravel() ax = ax.ravel().tolist() _validate_if_list_of_axes(ax_flat) _validate_if_list_of_axes(ax_flat, 4) assert_raises(ValueError, _validate_if_list_of_axes, ax_flat, 5) assert_raises(ValueError, _validate_if_list_of_axes, ax, 3) assert_raises(ValueError, _validate_if_list_of_axes, 'error') assert_raises(ValueError, _validate_if_list_of_axes, ['error'] * 2) assert_raises(ValueError, _validate_if_list_of_axes, ax[0]) assert_raises(ValueError, _validate_if_list_of_axes, ax, 3) ax_flat[2] = 23 assert_raises(ValueError, _validate_if_list_of_axes, ax_flat) _validate_if_list_of_axes(ax, 4)
def test_validate_if_list_of_axes(): """Test validation of axes.""" fig, ax = plt.subplots(2, 2) pytest.raises(ValueError, _validate_if_list_of_axes, ax) ax_flat = ax.ravel() ax = ax.ravel().tolist() _validate_if_list_of_axes(ax_flat) _validate_if_list_of_axes(ax_flat, 4) pytest.raises(ValueError, _validate_if_list_of_axes, ax_flat, 5) pytest.raises(ValueError, _validate_if_list_of_axes, ax, 3) pytest.raises(ValueError, _validate_if_list_of_axes, 'error') pytest.raises(ValueError, _validate_if_list_of_axes, ['error'] * 2) pytest.raises(ValueError, _validate_if_list_of_axes, ax[0]) pytest.raises(ValueError, _validate_if_list_of_axes, ax, 3) ax_flat[2] = 23 pytest.raises(ValueError, _validate_if_list_of_axes, ax_flat) _validate_if_list_of_axes(ax, 4) plt.close('all')
def test_validate_if_list_of_axes(): """Test validation of axes.""" import matplotlib.pyplot as plt fig, ax = plt.subplots(2, 2) assert_raises(ValueError, _validate_if_list_of_axes, ax) ax_flat = ax.ravel() ax = ax.ravel().tolist() _validate_if_list_of_axes(ax_flat) _validate_if_list_of_axes(ax_flat, 4) assert_raises(ValueError, _validate_if_list_of_axes, ax_flat, 5) assert_raises(ValueError, _validate_if_list_of_axes, ax, 3) assert_raises(ValueError, _validate_if_list_of_axes, 'error') assert_raises(ValueError, _validate_if_list_of_axes, ['error'] * 2) assert_raises(ValueError, _validate_if_list_of_axes, ax[0]) assert_raises(ValueError, _validate_if_list_of_axes, ax, 3) ax_flat[2] = 23 assert_raises(ValueError, _validate_if_list_of_axes, ax_flat) _validate_if_list_of_axes(ax, 4)