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')