From 2ac2db258f915a971f4f18c24bcfc7c6c6e27be2 Mon Sep 17 00:00:00 2001 From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com> Date: Sun, 6 Jul 2025 00:38:22 +0100 Subject: [PATCH] Add tests for matrix multiplication and transpose operations in Frame --- src/frame/ops.rs | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/frame/ops.rs b/src/frame/ops.rs index ef23ca8..dea2396 100644 --- a/src/frame/ops.rs +++ b/src/frame/ops.rs @@ -135,6 +135,17 @@ mod tests { ); assert_eq!(frame.is_nan(), frame.matrix().is_nan()); assert_eq!(frame.apply_axis(Axis::Row, |x| x[0] + x[1]), vec![4.0, 6.0]); + + + assert_eq!(frame.matrix_mul(&frame), frame.matrix().matrix_mul(&frame.matrix())); + assert_eq!(frame.dot(&frame), frame.matrix().dot(&frame.matrix())); + + // test transpose - returns a matrix. + let frame_transposed_mat = frame.transpose(); + let frame_mat_transposed = frame.matrix().transpose(); + assert_eq!(frame_transposed_mat, frame_mat_transposed); + assert_eq!(frame.matrix(), &frame.matrix().transpose().transpose()); + } #[test] @@ -159,4 +170,7 @@ mod tests { vec![false, false] ); } + + + }