From 6804e7ca5a73e5f3e6271d1e4fa9b5e9e0230acc Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Sat, 26 Apr 2025 02:42:38 +0100
Subject: [PATCH] Add unit test for mutable data access in Matrix

---
 src/matrix/mat.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/matrix/mat.rs b/src/matrix/mat.rs
index 5a003d4..20ff191 100644
--- a/src/matrix/mat.rs
+++ b/src/matrix/mat.rs
@@ -613,6 +613,22 @@ mod tests {
         assert_eq!(rows[1], vec![2, 4, 6, 8]);
     }
 
+    // test data_mut
+    #[test]
+    fn test_data_mut() {
+        let mut matrix = create_test_matrix(); // 3x3
+        // 1 4 7
+        // 2 5 8
+        // 3 6 9
+
+        let data_mut = matrix.data_mut();
+        data_mut[0] = 10;
+        data_mut[1] = 20;
+
+        assert_eq!(matrix[(0, 0)], 10);
+        assert_eq!(matrix[(1, 0)], 20);
+    }
+
     #[test]
     fn test_matrix_row_get_and_iter() {
         let matrix = create_test_matrix_2x4(); // 2x4