use crate::{number::c64, DiagonalMatrix};
use rayon::prelude::*;
impl DiagonalMatrix<f64> {
pub fn powi(self, exp: i32) -> Self {
let mut slf = self;
slf.d.par_iter_mut().for_each(|di| *di = di.powi(exp));
slf
}
}
impl DiagonalMatrix<c64> {
pub fn powi(self, exp: i32) -> Self {
let mut slf = self;
slf.d.par_iter_mut().for_each(|di| *di = di.powi(exp));
slf
}
}
#[cfg(test)]
mod tests {
use crate::*;
#[test]
fn it_works() {
let a = DiagonalMatrix::<f64>::identity(2);
let a_inv = a.clone().powi(-1);
assert_eq!(a[0], a_inv[0]);
let a = DiagonalMatrix::<c64>::identity(2);
let a_inv = a.clone().powi(-1);
assert_eq!(a[0], a_inv[0]);
}
}