Add tests for Quaternion dunder method __str__().

Mayitzin · Nov 6, 2024 · aad2707 · aad2707
1 parent 6c269ab
commit aad2707
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_quaternions.py b/tests/test_quaternions.py
@@ -165,6 +165,11 @@ def test_log(self):
         np.testing.assert_allclose(ahrs.Quaternion([1, 0, 1, 0]).log, [0, 0, np.pi/4, 0])
         np.testing.assert_allclose(ahrs.Quaternion([1, 0, 0, 1]).log, [0, 0, 0, np.pi/4])
 
+    def test_str(self):
+        self.assertEqual(str(self.q0), "(1.0000 +0.0000i +0.0000j +0.0000k)")
+        self.assertEqual(str(self.q3), f"(0.0000 {self.vector3[0]:+.4f}i {self.vector3[1]:+.4f}j {self.vector3[2]:+.4f}k)")
+        self.assertEqual(str(self.q4), f"({self.vector4[0]:-.4f} {self.vector4[1]:+.4f}i {self.vector4[2]:+.4f}j {self.vector4[3]:+.4f}k)")
+
 class TestQuaternionArray(unittest.TestCase):
     def setUp(self) -> None:
         self.Q0 = ahrs.QuaternionArray()