output type stick to python float

immaxchen · immaxchen · commit 230f0fb567e0 · 2020-06-10T21:49:11.000+08:00
diff --git a/lib/matplotlib/tests/test_text.py b/lib/matplotlib/tests/test_text.py
@@ -13,6 +13,7 @@
 import matplotlib.transforms as mtransforms
 from matplotlib.testing.decorators import check_figures_equal, image_comparison
 
+
 needs_usetex = pytest.mark.skipif(
     not mpl.checkdep_usetex(True),
     reason="This test needs a TeX installation")
@@ -696,5 +697,5 @@ def test_transform_rotates_text():
     transform = mtransforms.Affine2D().rotate_deg(30)
     text = ax.text(0, 0, 'test', transform=transform,
                    transform_rotates_text=True)
-    result = text.get_rotation()[0]
+    result = text.get_rotation()
     assert_almost_equal(result, 30)
diff --git a/lib/matplotlib/text.py b/lib/matplotlib/text.py
@@ -233,7 +233,8 @@ def get_rotation(self):
         if self.get_transform_rotates_text():
             angle = get_rotation(self._rotation)
             x, y = self.get_unitless_position()
-            return self.get_transform().transform_angles([angle, ], [[x, y]])
+            trans = self.get_transform()
+            return trans.transform_angles([angle], [[x, y]]).item(0)
         else:
             return get_rotation(self._rotation)  # string_or_number -> number
 
