Add test that polygon tuple markers work.

QuLogic · QuLogic · commit df3ff69ade03 · 2020-03-14T04:23:23.000-04:00
diff --git a/lib/matplotlib/tests/test_marker.py b/lib/matplotlib/tests/test_marker.py
@@ -1,6 +1,7 @@
 import numpy as np
 from matplotlib import markers
 from matplotlib.path import Path
+from matplotlib.testing.decorators import check_figures_equal
 
 import pytest
 
@@ -26,3 +27,61 @@ def test_marker_path():
     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*.
+    size = 20
+
+    # Squares
+    ax_test.plot([0] * 3, marker=(4, 0, 45), markersize=size)
+    ax_ref.plot([0] * 3, marker='s', markersize=size/np.sqrt(2))
+
+    # Diamonds, with and without rotation argument
+    ax_test.plot([1] * 3, marker=(4, 0), markersize=size)
+    ax_ref.plot([1] * 3, marker=UnsnappedMarkerStyle('D'),
+                markersize=size/np.sqrt(2))
+    ax_test.plot([1.5] * 3, marker=(4, 0, 0), markersize=size)
+    ax_ref.plot([1.5] * 3, marker=UnsnappedMarkerStyle('D'),
+                markersize=size/np.sqrt(2))
+
+    # Pentagon, with and without rotation argument
+    ax_test.plot([2] * 3, marker=(5, 0), markersize=size)
+    ax_ref.plot([2] * 3, marker=UnsnappedMarkerStyle('p'), markersize=size)
+    ax_test.plot([2.5] * 3, marker=(5, 0, 0), markersize=size)
+    ax_ref.plot([2.5] * 3, marker=UnsnappedMarkerStyle('p'), markersize=size)
+
+    # Hexagon, with and without rotation argument
+    ax_test.plot([3] * 3, marker=(6, 0), markersize=size)
+    ax_ref.plot([3] * 3, marker='h', markersize=size)
+    ax_test.plot([3.5] * 3, marker=(6, 0, 0), markersize=size)
+    ax_ref.plot([3.5] * 3, marker='h', markersize=size)
+
+    # Rotated hexagon
+    ax_test.plot([4] * 3, marker=(6, 0, 30), markersize=size)
+    ax_ref.plot([4] * 3, marker='H', markersize=size)
+
+    # Octagons
+    ax_test.plot([5] * 3, marker=(8, 0, 22.5), markersize=size)
+    ax_ref.plot([5] * 3, marker=UnsnappedMarkerStyle('8'), markersize=size)
+
+    ax_test.set(xlim=(-0.5, 2.5), ylim=(-0.5, 5.5))
+    ax_ref.set(xlim=(-0.5, 2.5), ylim=(-0.5, 5.5))
