Added handle position checks to test_widgets.py

andrew-fennell · andrew-fennell · commit 787db12a0c6f · 2022-03-28T10:00:10.000-05:00
diff --git a/lib/matplotlib/tests/test_widgets.py b/lib/matplotlib/tests/test_widgets.py
@@ -1107,17 +1107,38 @@ def test_range_slider(orientation):
 
     slider.set_val((0.2, 0.6))
     assert_allclose(slider.val, (0.2, 0.6))
+    if orientation == "vertical":
+        hpositions = [h.get_ydata()[0] for h in slider._handles]
+    else:
+        hpositions = [h.get_xdata()[0] for h in slider._handles]
+    assert_allclose(hpositions, (0.2, 0.6))
+
     box = slider.poly.get_extents().transformed(ax.transAxes.inverted())
     assert_allclose(box.get_points().flatten()[idx], [0.2, .25, 0.6, .75])
 
     slider.set_val((0.2, 0.1))
     assert_allclose(slider.val, (0.1, 0.2))
+    if orientation == "vertical":
+        hpositions = [h.get_ydata()[0] for h in slider._handles]
+    else:
+        hpositions = [h.get_xdata()[0] for h in slider._handles]
+    assert_allclose(hpositions, (0.1, 0.2))
 
     slider.set_val((-1, 10))
     assert_allclose(slider.val, (0, 1))
+    if orientation == "vertical":
+        hpositions = [h.get_ydata()[0] for h in slider._handles]
+    else:
+        hpositions = [h.get_xdata()[0] for h in slider._handles]
+    assert_allclose(hpositions, (0, 1))
 
     slider.reset()
-    assert_allclose(slider.val, [0.1, 0.34])
+    assert_allclose(slider.val, (0.1, 0.34))
+    if orientation == "vertical":
+        hpositions = [h.get_ydata()[0] for h in slider._handles]
+    else:
+        hpositions = [h.get_xdata()[0] for h in slider._handles]
+    assert_allclose(hpositions, (0.1, 0.34))
 
 
 def check_polygon_selector(event_sequence, expected_result, selections_count,
