Add tests for matrix multiplication and transpose operations in Frame

This commit is contained in:
Palash Tyagi 2025-07-06 00:38:22 +01:00
parent 7a68d13eb3
commit 2ac2db258f

View File

@ -135,6 +135,17 @@ mod tests {
); );
assert_eq!(frame.is_nan(), frame.matrix().is_nan()); 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.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] #[test]
@ -159,4 +170,7 @@ mod tests {
vec![false, false] vec![false, false]
); );
} }
} }