Add tests catching arrays with different shapes.

Mayitzin · Oct 23, 2024 · e108694 · e108694
1 parent 238d4b8
commit e108694
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_estimators.py b/tests/test_estimators.py
@@ -291,6 +291,10 @@ def test_wrong_input_vectors(self):
         self.assertRaises(TypeError, ahrs.filters.AQUA, acc=True, mag=[1.0, 2.0, 3.0])
         self.assertRaises(ValueError, ahrs.filters.AQUA, acc=[1.0, 2.0], mag=[2.0, 3.0, 4.0])
         self.assertRaises(ValueError, ahrs.filters.AQUA, acc=[1.0, 2.0, 3.0, 4.0], mag=[2.0, 3.0, 4.0, 5.0])
+        self.assertRaises(TypeError, ahrs.filters.AQUA, acc='self.accelerometers', mag=self.magnetometers)
+        self.assertRaises(TypeError, ahrs.filters.AQUA, acc=self.accelerometers, mag='self.magnetometers')
+        self.assertRaises(ValueError, ahrs.filters.AQUA, acc=[[1., 2., 3.]], mag=self.magnetometers)
+        self.assertRaises(ValueError, ahrs.filters.AQUA, acc=self.accelerometers, mag=[[1., 2., 3.]])
 
     def test_wrong_input_vector_types(self):
         self.assertRaises(TypeError, ahrs.filters.AQUA, acc=['1.0', 2.0, 3.0], mag=[2.0, 3.0, 4.0])