common: Add asserts for rotate_*

pedantic79 · Jan 3, 2024 · 9c4100a · 9c4100a
1 parent 078dccb
commit 9c4100a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/common/matrix.rs b/src/common/matrix.rs
@@ -3,6 +3,7 @@ where
     T: Default + Copy,
     A: AsMut<[T]>,
 {
+    assert_eq!(a.len(), a[0].as_mut().len(), "not a square matrix");
     let len = a.len();
     for i in 0..(len / 2) {
         for j in i..(len - i - 1) {
@@ -20,6 +21,7 @@ where
     T: Default + Copy,
     A: AsMut<[T]>,
 {
+    assert_eq!(a.len(), a[0].as_mut().len(), "not a square matrix");
     let len = a.len();
     for i in 0..(len / 2) {
         for j in i..(len - i - 1) {
@@ -37,6 +39,7 @@ where
     T: Default + Copy,
     A: AsMut<[T]>,
 {
+    assert_eq!(a.len(), a[0].as_mut().len(), "not a square matrix");
     let len = a.len();
 
     if len % 2 == 1 {
@@ -59,6 +62,7 @@ where
     T: Default + Copy,
     A: AsMut<[T]>,
 {
+    assert_eq!(a.len(), a[0].as_mut().len(), "not a square matrix");
     for row in a.iter_mut() {
         row.as_mut().reverse();
     }