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)
