forked from 170010011/fr
280 lines
8.8 KiB
Python
280 lines
8.8 KiB
Python
|
import io
|
||
|
|
from itertools import chain
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
import matplotlib.patches as mpatches
|
||
|
|
import matplotlib.lines as mlines
|
||
|
|
import matplotlib.path as mpath
|
||
|
|
import matplotlib.transforms as mtransforms
|
||
|
|
import matplotlib.collections as mcollections
|
||
|
|
import matplotlib.artist as martist
|
||
|
|
from matplotlib.testing.decorators import image_comparison
|
||
|
|
|
||
|
|
|
||
|
|
def test_patch_transform_of_none():
|
||
|
|
# tests the behaviour of patches added to an Axes with various transform
|
||
|
|
# specifications
|
||
|
|
|
||
|
|
ax = plt.axes()
|
||
|
|
ax.set_xlim([1, 3])
|
||
|
|
ax.set_ylim([1, 3])
|
||
|
|
|
||
|
|
# Draw an ellipse over data coord (2, 2) by specifying device coords.
|
||
|
|
xy_data = (2, 2)
|
||
|
|
xy_pix = ax.transData.transform(xy_data)
|
||
|
|
|
||
|
|
# Not providing a transform of None puts the ellipse in data coordinates .
|
||
|
|
e = mpatches.Ellipse(xy_data, width=1, height=1, fc='yellow', alpha=0.5)
|
||
|
|
ax.add_patch(e)
|
||
|
|
assert e._transform == ax.transData
|
||
|
|
|
||
|
|
# Providing a transform of None puts the ellipse in device coordinates.
|
||
|
|
e = mpatches.Ellipse(xy_pix, width=120, height=120, fc='coral',
|
||
|
|
transform=None, alpha=0.5)
|
||
|
|
assert e.is_transform_set()
|
||
|
|
ax.add_patch(e)
|
||
|
|
assert isinstance(e._transform, mtransforms.IdentityTransform)
|
||
|
|
|
||
|
|
# Providing an IdentityTransform puts the ellipse in device coordinates.
|
||
|
|
e = mpatches.Ellipse(xy_pix, width=100, height=100,
|
||
|
|
transform=mtransforms.IdentityTransform(), alpha=0.5)
|
||
|
|
ax.add_patch(e)
|
||
|
|
assert isinstance(e._transform, mtransforms.IdentityTransform)
|
||
|
|
|
||
|
|
# Not providing a transform, and then subsequently "get_transform" should
|
||
|
|
# not mean that "is_transform_set".
|
||
|
|
e = mpatches.Ellipse(xy_pix, width=120, height=120, fc='coral',
|
||
|
|
alpha=0.5)
|
||
|
|
intermediate_transform = e.get_transform()
|
||
|
|
assert not e.is_transform_set()
|
||
|
|
ax.add_patch(e)
|
||
|
|
assert e.get_transform() != intermediate_transform
|
||
|
|
assert e.is_transform_set()
|
||
|
|
assert e._transform == ax.transData
|
||
|
|
|
||
|
|
|
||
|
|
def test_collection_transform_of_none():
|
||
|
|
# tests the behaviour of collections added to an Axes with various
|
||
|
|
# transform specifications
|
||
|
|
|
||
|
|
ax = plt.axes()
|
||
|
|
ax.set_xlim([1, 3])
|
||
|
|
ax.set_ylim([1, 3])
|
||
|
|
|
||
|
|
# draw an ellipse over data coord (2, 2) by specifying device coords
|
||
|
|
xy_data = (2, 2)
|
||
|
|
xy_pix = ax.transData.transform(xy_data)
|
||
|
|
|
||
|
|
# not providing a transform of None puts the ellipse in data coordinates
|
||
|
|
e = mpatches.Ellipse(xy_data, width=1, height=1)
|
||
|
|
c = mcollections.PatchCollection([e], facecolor='yellow', alpha=0.5)
|
||
|
|
ax.add_collection(c)
|
||
|
|
# the collection should be in data coordinates
|
||
|
|
assert c.get_offset_transform() + c.get_transform() == ax.transData
|
||
|
|
|
||
|
|
# providing a transform of None puts the ellipse in device coordinates
|
||
|
|
e = mpatches.Ellipse(xy_pix, width=120, height=120)
|
||
|
|
c = mcollections.PatchCollection([e], facecolor='coral',
|
||
|
|
alpha=0.5)
|
||
|
|
c.set_transform(None)
|
||
|
|
ax.add_collection(c)
|
||
|
|
assert isinstance(c.get_transform(), mtransforms.IdentityTransform)
|
||
|
|
|
||
|
|
# providing an IdentityTransform puts the ellipse in device coordinates
|
||
|
|
e = mpatches.Ellipse(xy_pix, width=100, height=100)
|
||
|
|
c = mcollections.PatchCollection([e],
|
||
|
|
transform=mtransforms.IdentityTransform(),
|
||
|
|
alpha=0.5)
|
||
|
|
ax.add_collection(c)
|
||
|
|
assert isinstance(c._transOffset, mtransforms.IdentityTransform)
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(["clip_path_clipping"], remove_text=True)
|
||
|
|
def test_clipping():
|
||
|
|
exterior = mpath.Path.unit_rectangle().deepcopy()
|
||
|
|
exterior.vertices *= 4
|
||
|
|
exterior.vertices -= 2
|
||
|
|
interior = mpath.Path.unit_circle().deepcopy()
|
||
|
|
interior.vertices = interior.vertices[::-1]
|
||
|
|
clip_path = mpath.Path.make_compound_path(exterior, interior)
|
||
|
|
|
||
|
|
star = mpath.Path.unit_regular_star(6).deepcopy()
|
||
|
|
star.vertices *= 2.6
|
||
|
|
|
||
|
|
ax1 = plt.subplot(121)
|
||
|
|
col = mcollections.PathCollection([star], lw=5, edgecolor='blue',
|
||
|
|
facecolor='red', alpha=0.7, hatch='*')
|
||
|
|
col.set_clip_path(clip_path, ax1.transData)
|
||
|
|
ax1.add_collection(col)
|
||
|
|
|
||
|
|
ax2 = plt.subplot(122, sharex=ax1, sharey=ax1)
|
||
|
|
patch = mpatches.PathPatch(star, lw=5, edgecolor='blue', facecolor='red',
|
||
|
|
alpha=0.7, hatch='*')
|
||
|
|
patch.set_clip_path(clip_path, ax2.transData)
|
||
|
|
ax2.add_patch(patch)
|
||
|
|
|
||
|
|
ax1.set_xlim([-3, 3])
|
||
|
|
ax1.set_ylim([-3, 3])
|
||
|
|
|
||
|
|
|
||
|
|
def test_cull_markers():
|
||
|
|
x = np.random.random(20000)
|
||
|
|
y = np.random.random(20000)
|
||
|
|
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
ax.plot(x, y, 'k.')
|
||
|
|
ax.set_xlim(2, 3)
|
||
|
|
|
||
|
|
pdf = io.BytesIO()
|
||
|
|
fig.savefig(pdf, format="pdf")
|
||
|
|
assert len(pdf.getvalue()) < 8000
|
||
|
|
|
||
|
|
svg = io.BytesIO()
|
||
|
|
fig.savefig(svg, format="svg")
|
||
|
|
assert len(svg.getvalue()) < 20000
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(['hatching'], remove_text=True, style='default')
|
||
|
|
def test_hatching():
|
||
|
|
fig, ax = plt.subplots(1, 1)
|
||
|
|
|
||
|
|
# Default hatch color.
|
||
|
|
rect1 = mpatches.Rectangle((0, 0), 3, 4, hatch='/')
|
||
|
|
ax.add_patch(rect1)
|
||
|
|
|
||
|
|
rect2 = mcollections.RegularPolyCollection(4, sizes=[16000],
|
||
|
|
offsets=[(1.5, 6.5)],
|
||
|
|
transOffset=ax.transData,
|
||
|
|
hatch='/')
|
||
|
|
ax.add_collection(rect2)
|
||
|
|
|
||
|
|
# Ensure edge color is not applied to hatching.
|
||
|
|
rect3 = mpatches.Rectangle((4, 0), 3, 4, hatch='/', edgecolor='C1')
|
||
|
|
ax.add_patch(rect3)
|
||
|
|
|
||
|
|
rect4 = mcollections.RegularPolyCollection(4, sizes=[16000],
|
||
|
|
offsets=[(5.5, 6.5)],
|
||
|
|
transOffset=ax.transData,
|
||
|
|
hatch='/', edgecolor='C1')
|
||
|
|
ax.add_collection(rect4)
|
||
|
|
|
||
|
|
ax.set_xlim(0, 7)
|
||
|
|
ax.set_ylim(0, 9)
|
||
|
|
|
||
|
|
|
||
|
|
def test_remove():
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
im = ax.imshow(np.arange(36).reshape(6, 6))
|
||
|
|
ln, = ax.plot(range(5))
|
||
|
|
|
||
|
|
assert fig.stale
|
||
|
|
assert ax.stale
|
||
|
|
|
||
|
|
fig.canvas.draw()
|
||
|
|
assert not fig.stale
|
||
|
|
assert not ax.stale
|
||
|
|
assert not ln.stale
|
||
|
|
|
||
|
|
assert im in ax._mouseover_set
|
||
|
|
assert ln not in ax._mouseover_set
|
||
|
|
assert im.axes is ax
|
||
|
|
|
||
|
|
im.remove()
|
||
|
|
ln.remove()
|
||
|
|
|
||
|
|
for art in [im, ln]:
|
||
|
|
assert art.axes is None
|
||
|
|
assert art.figure is None
|
||
|
|
|
||
|
|
assert im not in ax._mouseover_set
|
||
|
|
assert fig.stale
|
||
|
|
assert ax.stale
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(["default_edges.png"], remove_text=True, style='default')
|
||
|
|
def test_default_edges():
|
||
|
|
# Remove this line when this test image is regenerated.
|
||
|
|
plt.rcParams['text.kerning_factor'] = 6
|
||
|
|
|
||
|
|
fig, [[ax1, ax2], [ax3, ax4]] = plt.subplots(2, 2)
|
||
|
|
|
||
|
|
ax1.plot(np.arange(10), np.arange(10), 'x',
|
||
|
|
np.arange(10) + 1, np.arange(10), 'o')
|
||
|
|
ax2.bar(np.arange(10), np.arange(10), align='edge')
|
||
|
|
ax3.text(0, 0, "BOX", size=24, bbox=dict(boxstyle='sawtooth'))
|
||
|
|
ax3.set_xlim((-1, 1))
|
||
|
|
ax3.set_ylim((-1, 1))
|
||
|
|
pp1 = mpatches.PathPatch(
|
||
|
|
mpath.Path([(0, 0), (1, 0), (1, 1), (0, 0)],
|
||
|
|
[mpath.Path.MOVETO, mpath.Path.CURVE3,
|
||
|
|
mpath.Path.CURVE3, mpath.Path.CLOSEPOLY]),
|
||
|
|
fc="none", transform=ax4.transData)
|
||
|
|
ax4.add_patch(pp1)
|
||
|
|
|
||
|
|
|
||
|
|
def test_properties():
|
||
|
|
ln = mlines.Line2D([], [])
|
||
|
|
ln.properties() # Check that no warning is emitted.
|
||
|
|
|
||
|
|
|
||
|
|
def test_setp():
|
||
|
|
# Check empty list
|
||
|
|
plt.setp([])
|
||
|
|
plt.setp([[]])
|
||
|
|
|
||
|
|
# Check arbitrary iterables
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
lines1 = ax.plot(range(3))
|
||
|
|
lines2 = ax.plot(range(3))
|
||
|
|
martist.setp(chain(lines1, lines2), 'lw', 5)
|
||
|
|
plt.setp(ax.spines.values(), color='green')
|
||
|
|
|
||
|
|
# Check *file* argument
|
||
|
|
sio = io.StringIO()
|
||
|
|
plt.setp(lines1, 'zorder', file=sio)
|
||
|
|
assert sio.getvalue() == ' zorder: float\n'
|
||
|
|
|
||
|
|
|
||
|
|
def test_None_zorder():
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
ln, = ax.plot(range(5), zorder=None)
|
||
|
|
assert ln.get_zorder() == mlines.Line2D.zorder
|
||
|
|
ln.set_zorder(123456)
|
||
|
|
assert ln.get_zorder() == 123456
|
||
|
|
ln.set_zorder(None)
|
||
|
|
assert ln.get_zorder() == mlines.Line2D.zorder
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('accept_clause, expected', [
|
||
|
|
('', 'unknown'),
|
||
|
|
("ACCEPTS: [ '-' | '--' | '-.' ]", "[ '-' | '--' | '-.' ]"),
|
||
|
|
('ACCEPTS: Some description.', 'Some description.'),
|
||
|
|
('.. ACCEPTS: Some description.', 'Some description.'),
|
||
|
|
('arg : int', 'int'),
|
||
|
|
('*arg : int', 'int'),
|
||
|
|
('arg : int\nACCEPTS: Something else.', 'Something else. '),
|
||
|
|
])
|
||
|
|
def test_artist_inspector_get_valid_values(accept_clause, expected):
|
||
|
|
class TestArtist(martist.Artist):
|
||
|
|
def set_f(self, arg):
|
||
|
|
pass
|
||
|
|
|
||
|
|
TestArtist.set_f.__doc__ = """
|
||
|
|
Some text.
|
||
|
|
|
||
|
|
%s
|
||
|
|
""" % accept_clause
|
||
|
|
valid_values = martist.ArtistInspector(TestArtist).get_valid_values('f')
|
||
|
|
assert valid_values == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_artist_inspector_get_aliases():
|
||
|
|
# test the correct format and type of get_aliases method
|
||
|
|
ai = martist.ArtistInspector(mlines.Line2D)
|
||
|
|
aliases = ai.get_aliases()
|
||
|
|
assert aliases["linewidth"] == {"lw"}
|