Rotate around test wip [skip ci]

trananso · trananso · commit 492d16c68f82 · 2024-04-18T17:05:41.000-04:00
diff --git a/lib/matplotlib/tests/test_transforms.py b/lib/matplotlib/tests/test_transforms.py
@@ -426,10 +426,54 @@ def test_rotate(self):
             [0, -2, 0], [3, 0, 0], [0, 0, 4], [5, -5, 0], [6, -6, 6]])
 
         for dim in range(3):
-            assert_array_equal((r90[dim] + r90[dim]).get_matrix(),
-                               r180[dim].get_matrix())
-            assert_array_equal((r90[dim] + r180[dim]).get_matrix(),
-                               r270[dim].get_matrix())
+            assert_array_almost_equal(
+                (r90[dim] + r90[dim]).get_matrix(), r180[dim].get_matrix())
+            assert_array_almost_equal(
+                (r90[dim] + r180[dim]).get_matrix(), r270[dim].get_matrix())
+
+    def test_rotate_around(self):
+        r_pi_2 = [Affine3D().rotate_around(*self.pivot, np.pi / 2, dim)
+                  for dim in range(3)]
+        r90 = [Affine3D().rotate_deg_around(*self.pivot, 90, dim) for dim in range(3)]
+
+        for dim in range(3):
+            assert_array_equal(r_pi_2[dim].get_matrix(), r90[dim].get_matrix())
+
+        # assert_array_almost_equal(r90.transform(self.single_point), [1, 1])
+        # assert_array_almost_equal(r90.transform(self.multiple_points),
+        #                           [[0, 0], [-1, 3], [2, 4]])
+
+        r_pi = [Affine3D().rotate_around(*self.pivot, np.pi, dim) for dim in range(3)]
+        r180 = [Affine3D().rotate_deg_around(*self.pivot, 180, dim) for dim in range(3)]
+
+        for dim in range(3):
+            assert_array_equal(r_pi[dim].get_matrix(), r180[dim].get_matrix())
+
+        # assert_array_almost_equal(r180.transform(self.single_point), [1, 1])
+        # assert_array_almost_equal(r180.transform(self.multiple_points),
+        #                           [[2, 0], [-1, -1], [-2, 2]])
+
+        r_pi_3_2 = [Affine3D().rotate_around(*self.pivot, 3 * np.pi / 2, dim)
+                    for dim in range(3)]
+        r270 = [Affine3D().rotate_deg_around(*self.pivot, 270, dim) for dim in range(3)]
+
+        for dim in range(3):
+            assert_array_equal(r_pi_3_2[dim].get_matrix(), r270[dim].get_matrix())
+
+        # assert_array_almost_equal(r270.transform(self.single_point), [1, 1])
+        # assert_array_almost_equal(r270.transform(self.multiple_points),
+        #                           [[2, 2], [3, -1], [0, -2]])
+
+        # raise TypeError(np.array_str((r90[2]+r90[2]).get_matrix(),
+        #                precision=2, suppress_small=True))
+        # raise TypeError(np.array_str(r180[2].get_matrix(),
+        #                              precision=2, suppress_small=True))
+
+        for dim in range(3):
+            assert_array_almost_equal(
+                (r90[dim] + r90[dim]).get_matrix(), r180[dim].get_matrix())
+            assert_array_almost_equal(
+                (r90[dim] + r180[dim]).get_matrix(), r270[dim].get_matrix())
 
 
 def test_non_affine_caching():
