mirror of
https://github.com/Magnus167/rustframe.git
synced 2025-08-20 04:00:01 +00:00
Add tests for matrix multiplication and transpose operations in Frame
This commit is contained in:
parent
7a68d13eb3
commit
2ac2db258f
@ -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]
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user