add more tests for angle wrapping

petercorke · petercorke · commit c234918e97df · 2023-03-29T08:03:40.000+10:00
diff --git a/tests/base/test_vectors.py b/tests/base/test_vectors.py
@@ -230,6 +230,11 @@ def test_angdiff(self):
         self.assertEqual(angdiff(-pi, pi), 0)
 
         nt.assert_array_almost_equal(angdiff([0, -pi, pi], 0), [0, -pi, -pi])
+        nt.assert_array_almost_equal(angdiff([0, -pi, pi], pi), [-pi, 0, 0])
+
+        nt.assert_array_almost_equal(angdiff(0, [0, -pi, pi]), [0, -pi, -pi])
+        nt.assert_array_almost_equal(angdiff(pi, [0, -pi, pi]), [-pi, 0, 0])
+
         nt.assert_array_almost_equal(angdiff([1, 2, 3], [1, 2, 3]), [0, 0, 0])
 
     def test_wrap(self):
