177 lines
6.3 KiB
Python
177 lines
6.3 KiB
Python
|
import numpy as np
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
from matplotlib import markers
|
||
|
|
from matplotlib.path import Path
|
||
|
|
from matplotlib.testing.decorators import check_figures_equal
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_markers_valid():
|
||
|
|
marker_style = markers.MarkerStyle()
|
||
|
|
mrk_array = np.array([[-0.5, 0],
|
||
|
|
[0.5, 0]])
|
||
|
|
# Checking this doesn't fail.
|
||
|
|
marker_style.set_marker(mrk_array)
|
||
|
|
|
||
|
|
|
||
|
|
def test_markers_invalid():
|
||
|
|
marker_style = markers.MarkerStyle()
|
||
|
|
mrk_array = np.array([[-0.5, 0, 1, 2, 3]])
|
||
|
|
# Checking this does fail.
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
marker_style.set_marker(mrk_array)
|
||
|
|
|
||
|
|
|
||
|
|
def test_marker_path():
|
||
|
|
marker_style = markers.MarkerStyle()
|
||
|
|
path = Path([[0, 0], [1, 0]], [Path.MOVETO, Path.LINETO])
|
||
|
|
# Checking this doesn't fail.
|
||
|
|
marker_style.set_marker(path)
|
||
|
|
|
||
|
|
|
||
|
|
class UnsnappedMarkerStyle(markers.MarkerStyle):
|
||
|
|
"""
|
||
|
|
A MarkerStyle where the snap threshold is force-disabled.
|
||
|
|
|
||
|
|
This is used to compare to polygon/star/asterisk markers which do not have
|
||
|
|
any snap threshold set.
|
||
|
|
"""
|
||
|
|
def _recache(self):
|
||
|
|
super()._recache()
|
||
|
|
self._snap_threshold = None
|
||
|
|
|
||
|
|
|
||
|
|
@check_figures_equal()
|
||
|
|
def test_poly_marker(fig_test, fig_ref):
|
||
|
|
ax_test = fig_test.add_subplot()
|
||
|
|
ax_ref = fig_ref.add_subplot()
|
||
|
|
|
||
|
|
# Note, some reference sizes must be different because they have unit
|
||
|
|
# *length*, while polygon markers are inscribed in a circle of unit
|
||
|
|
# *radius*. This introduces a factor of np.sqrt(2), but since size is
|
||
|
|
# squared, that becomes 2.
|
||
|
|
size = 20**2
|
||
|
|
|
||
|
|
# Squares
|
||
|
|
ax_test.scatter([0], [0], marker=(4, 0, 45), s=size)
|
||
|
|
ax_ref.scatter([0], [0], marker='s', s=size/2)
|
||
|
|
|
||
|
|
# Diamonds, with and without rotation argument
|
||
|
|
ax_test.scatter([1], [1], marker=(4, 0), s=size)
|
||
|
|
ax_ref.scatter([1], [1], marker=UnsnappedMarkerStyle('D'), s=size/2)
|
||
|
|
ax_test.scatter([1], [1.5], marker=(4, 0, 0), s=size)
|
||
|
|
ax_ref.scatter([1], [1.5], marker=UnsnappedMarkerStyle('D'), s=size/2)
|
||
|
|
|
||
|
|
# Pentagon, with and without rotation argument
|
||
|
|
ax_test.scatter([2], [2], marker=(5, 0), s=size)
|
||
|
|
ax_ref.scatter([2], [2], marker=UnsnappedMarkerStyle('p'), s=size)
|
||
|
|
ax_test.scatter([2], [2.5], marker=(5, 0, 0), s=size)
|
||
|
|
ax_ref.scatter([2], [2.5], marker=UnsnappedMarkerStyle('p'), s=size)
|
||
|
|
|
||
|
|
# Hexagon, with and without rotation argument
|
||
|
|
ax_test.scatter([3], [3], marker=(6, 0), s=size)
|
||
|
|
ax_ref.scatter([3], [3], marker='h', s=size)
|
||
|
|
ax_test.scatter([3], [3.5], marker=(6, 0, 0), s=size)
|
||
|
|
ax_ref.scatter([3], [3.5], marker='h', s=size)
|
||
|
|
|
||
|
|
# Rotated hexagon
|
||
|
|
ax_test.scatter([4], [4], marker=(6, 0, 30), s=size)
|
||
|
|
ax_ref.scatter([4], [4], marker='H', s=size)
|
||
|
|
|
||
|
|
# Octagons
|
||
|
|
ax_test.scatter([5], [5], marker=(8, 0, 22.5), s=size)
|
||
|
|
ax_ref.scatter([5], [5], marker=UnsnappedMarkerStyle('8'), s=size)
|
||
|
|
|
||
|
|
ax_test.set(xlim=(-0.5, 5.5), ylim=(-0.5, 5.5))
|
||
|
|
ax_ref.set(xlim=(-0.5, 5.5), ylim=(-0.5, 5.5))
|
||
|
|
|
||
|
|
|
||
|
|
def test_star_marker():
|
||
|
|
# We don't really have a strict equivalent to this marker, so we'll just do
|
||
|
|
# a smoke test.
|
||
|
|
size = 20**2
|
||
|
|
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
ax.scatter([0], [0], marker=(5, 1), s=size)
|
||
|
|
ax.scatter([1], [1], marker=(5, 1, 0), s=size)
|
||
|
|
ax.set(xlim=(-0.5, 0.5), ylim=(-0.5, 1.5))
|
||
|
|
|
||
|
|
|
||
|
|
# The asterisk marker is really a star with 0-size inner circle, so the ends
|
||
|
|
# are corners and get a slight bevel. The reference markers are just singular
|
||
|
|
# lines without corners, so they have no bevel, and we need to add a slight
|
||
|
|
# tolerance.
|
||
|
|
@check_figures_equal(tol=1.45)
|
||
|
|
def test_asterisk_marker(fig_test, fig_ref, request):
|
||
|
|
ax_test = fig_test.add_subplot()
|
||
|
|
ax_ref = fig_ref.add_subplot()
|
||
|
|
|
||
|
|
# Note, some reference sizes must be different because they have unit
|
||
|
|
# *length*, while asterisk markers are inscribed in a circle of unit
|
||
|
|
# *radius*. This introduces a factor of np.sqrt(2), but since size is
|
||
|
|
# squared, that becomes 2.
|
||
|
|
size = 20**2
|
||
|
|
|
||
|
|
def draw_ref_marker(y, style, size):
|
||
|
|
# As noted above, every line is doubled. Due to antialiasing, these
|
||
|
|
# doubled lines make a slight difference in the .png results.
|
||
|
|
ax_ref.scatter([y], [y], marker=UnsnappedMarkerStyle(style), s=size)
|
||
|
|
if request.getfixturevalue('ext') == 'png':
|
||
|
|
ax_ref.scatter([y], [y], marker=UnsnappedMarkerStyle(style),
|
||
|
|
s=size)
|
||
|
|
|
||
|
|
# Plus
|
||
|
|
ax_test.scatter([0], [0], marker=(4, 2), s=size)
|
||
|
|
draw_ref_marker(0, '+', size)
|
||
|
|
ax_test.scatter([0.5], [0.5], marker=(4, 2, 0), s=size)
|
||
|
|
draw_ref_marker(0.5, '+', size)
|
||
|
|
|
||
|
|
# Cross
|
||
|
|
ax_test.scatter([1], [1], marker=(4, 2, 45), s=size)
|
||
|
|
draw_ref_marker(1, 'x', size/2)
|
||
|
|
|
||
|
|
ax_test.set(xlim=(-0.5, 1.5), ylim=(-0.5, 1.5))
|
||
|
|
ax_ref.set(xlim=(-0.5, 1.5), ylim=(-0.5, 1.5))
|
||
|
|
|
||
|
|
|
||
|
|
@check_figures_equal()
|
||
|
|
def test_marker_clipping(fig_ref, fig_test):
|
||
|
|
# Plotting multiple markers can trigger different optimized paths in
|
||
|
|
# backends, so compare single markers vs multiple to ensure they are
|
||
|
|
# clipped correctly.
|
||
|
|
marker_count = len(markers.MarkerStyle.markers)
|
||
|
|
marker_size = 50
|
||
|
|
ncol = 7
|
||
|
|
nrow = marker_count // ncol + 1
|
||
|
|
|
||
|
|
width = 2 * marker_size * ncol
|
||
|
|
height = 2 * marker_size * nrow * 2
|
||
|
|
fig_ref.set_size_inches((width / fig_ref.dpi, height / fig_ref.dpi))
|
||
|
|
ax_ref = fig_ref.add_axes([0, 0, 1, 1])
|
||
|
|
fig_test.set_size_inches((width / fig_test.dpi, height / fig_ref.dpi))
|
||
|
|
ax_test = fig_test.add_axes([0, 0, 1, 1])
|
||
|
|
|
||
|
|
for i, marker in enumerate(markers.MarkerStyle.markers):
|
||
|
|
x = i % ncol
|
||
|
|
y = i // ncol * 2
|
||
|
|
|
||
|
|
# Singular markers per call.
|
||
|
|
ax_ref.plot([x, x], [y, y + 1], c='k', linestyle='-', lw=3)
|
||
|
|
ax_ref.plot(x, y, c='k',
|
||
|
|
marker=marker, markersize=marker_size, markeredgewidth=10,
|
||
|
|
fillstyle='full', markerfacecolor='white')
|
||
|
|
ax_ref.plot(x, y + 1, c='k',
|
||
|
|
marker=marker, markersize=marker_size, markeredgewidth=10,
|
||
|
|
fillstyle='full', markerfacecolor='white')
|
||
|
|
|
||
|
|
# Multiple markers in a single call.
|
||
|
|
ax_test.plot([x, x], [y, y + 1], c='k', linestyle='-', lw=3,
|
||
|
|
marker=marker, markersize=marker_size, markeredgewidth=10,
|
||
|
|
fillstyle='full', markerfacecolor='white')
|
||
|
|
|
||
|
|
ax_ref.set(xlim=(-0.5, ncol), ylim=(-0.5, 2 * nrow))
|
||
|
|
ax_test.set(xlim=(-0.5, ncol), ylim=(-0.5, 2 * nrow))
|
||
|
|
ax_ref.axis('off')
|
||
|
|
ax_test.axis('off')
|