Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

68 lines (55 sloc) 1.912 kb
import matplotlib
from nose.tools import assert_equal, assert_is, assert_is_not
from matplotlib.testing.decorators import image_comparison, cleanup
import matplotlib.pyplot as plt
@cleanup
def test_figure_label():
# pyplot figure creation, selection and closing with figure label and number
plt.close('all')
plt.figure('today')
plt.figure(3)
plt.figure('tomorrow')
plt.figure()
plt.figure(0)
plt.figure(1)
plt.figure(3)
assert_equal(plt.get_fignums(), [0, 1, 3, 4, 5])
assert_equal(plt.get_figlabels(), ['', 'today', '', 'tomorrow', ''])
plt.close(10)
plt.close()
plt.close(5)
plt.close('tomorrow')
assert_equal(plt.get_fignums(), [0, 1])
assert_equal(plt.get_figlabels(), ['', 'today'])
@image_comparison(baseline_images=['figure_today'])
def test_figure():
# named figure support
fig = plt.figure('today')
ax = fig.add_subplot(111)
ax.set_title(fig.get_label())
ax.plot(range(5))
# plot red line in a different figure.
plt.figure('tomorrow')
plt.plot([0, 1], [1,0], 'r')
# Return to the original; make sure the red line is not there.
plt.figure('today')
plt.close('tomorrow')
@cleanup
def test_gca():
fig = plt.figure()
ax1 = fig.add_axes([0, 0, 1, 1])
assert_is(fig.gca(projection='rectilinear'), ax1)
assert_is(fig.gca(), ax1)
ax2 = fig.add_subplot(121, projection='polar')
assert_is(fig.gca(), ax2)
assert_is(fig.gca(polar=True), ax2)
ax3 = fig.add_subplot(122)
assert_is(fig.gca(), ax3)
# the final request for a polar axes will end up creating one
# with a spec of 111.
assert_is_not(fig.gca(polar=True), ax3)
assert_is_not(fig.gca(polar=True), ax2)
assert_equal(fig.gca().get_geometry(), (1, 1, 1))
fig.sca(ax1)
assert_is(fig.gca(projection='rectilinear'), ax1)
assert_is(fig.gca(), ax1)
Jump to Line
Something went wrong with that request. Please try again.