Skip to content

Commit 3bb181a

Browse files
committed
Add test for arnoldi_householder
1 parent b33fd0d commit 3bb181a

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

src/krylov/arnoldi.rs

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -137,23 +137,22 @@ mod tests {
137137
use crate::{assert::*, generate::*};
138138

139139
#[test]
140-
fn aq_qh() {
141-
let a = array![[1.0, 2.0], [3.0, 5.0]];
142-
let mut v = Array::zeros(2);
143-
v[0] = 1.0;
140+
fn aq_qh_mgs() {
141+
let a: Array2<f64> = random((5, 5));
142+
let v: Array1<f64> = random(5);
144143
let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
145144
println!("A = \n{:?}", &a);
146145
println!("Q = \n{:?}", &q);
147146
println!("H = \n{:?}", &h);
148147
let aq = a.dot(&q);
149148
let qh = q.dot(&h);
150-
println!("AQ = \n{:?}", a.dot(&q));
151-
println!("QH = \n{:?}", q.dot(&h));
149+
println!("AQ = \n{:?}", &aq);
150+
println!("QH = \n{:?}", &qh);
152151
close_l2(&aq, &qh, 1e-9);
153152
}
154153

155154
#[test]
156-
fn aq_qh_random() {
155+
fn aq_qh_householder() {
157156
let a: Array2<f64> = random((5, 5));
158157
let v: Array1<f64> = random(5);
159158
let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
@@ -166,5 +165,4 @@ mod tests {
166165
println!("QH = \n{:?}", &qh);
167166
close_l2(&aq, &qh, 1e-9);
168167
}
169-
170168
}

0 commit comments

Comments
 (0)