def test_AxonMapSpatial_plot():
model = AxonMapSpatial()
ax = model.plot()
npt.assert_equal(isinstance(ax, Subplot), True)
# Electrodes and quadrants can be annotated:
for ann_q, n_q in [(True, 4), (False, 0)]:
ax = model.plot(annotate=ann_q)
npt.assert_equal(len(ax.texts), n_q)
close(ax.figure)
# Setting upside_down flips y axis:
ax = model.plot(upside_down=True, autoscale=True)
npt.assert_equal(ax.get_xlim(), (-5000, 5000))
npt.assert_equal(ax.get_ylim(), (4000, -4000))
def test_AxonMapSpatial_plot():
model = AxonMapSpatial()
for use_dva, xlim in zip([True, False], [(-18, 18), (-5000, 5000)]):
ax = model.plot(use_dva=use_dva)
npt.assert_equal(isinstance(ax, Subplot), True)
npt.assert_equal(ax.get_xlim(), xlim)
# Quadrants can be annotated:
for ann_q, n_q in [(True, 6), (False, 0)]:
fig, ax = plt.subplots()
model.plot(annotate=ann_q, ax=ax)
npt.assert_equal(len(ax.child_axes), int(n_q > 0))
if len(ax.child_axes) > 0:
npt.assert_equal(len(ax.child_axes[0].texts), n_q)
plt.close(fig)
def test_AxonMapSpatial_plot():
model = AxonMapSpatial()
for use_dva, xlim in zip([True, False], [(-18, 18), (-5000, 5000)]):
ax = model.plot(use_dva=use_dva)
npt.assert_equal(isinstance(ax, Subplot), True)
npt.assert_equal(ax.get_xlim(), xlim)
# Simulated area might be larger than that:
model = AxonMapSpatial(xrange=(-20.5, 20.5), yrange=(-16.1, 16.1))
ax = model.plot(use_dva=True)
npt.assert_almost_equal(ax.get_xlim(), (-21, 21))
npt.assert_almost_equal(ax.get_ylim(), (-18, 18))
ax = model.plot(use_dva=False)
npt.assert_almost_equal(ax.get_xlim(), (-6000, 6000))
npt.assert_almost_equal(ax.get_ylim(), (-5000, 5000))
# Figure size can be changed:
ax = model.plot(figsize=(8, 7))
npt.assert_almost_equal(ax.figure.get_size_inches(), (8, 7))
# Quadrants can be annotated:
for ann_q, n_q in [(True, 6), (False, 0)]:
fig, ax = plt.subplots()
model.plot(annotate=ann_q, ax=ax)
npt.assert_equal(len(ax.child_axes), int(n_q > 0))
if len(ax.child_axes) > 0:
npt.assert_equal(len(ax.child_axes[0].texts), n_q)
plt.close(fig)
