Skip to content

Commit 047ddf7

Browse files
committed
Add test
1 parent b02ee5b commit 047ddf7

File tree

1 file changed

+20
-2
lines changed

1 file changed

+20
-2
lines changed

tests/outer.rs

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,27 @@ include!("header.rs");
22
use ndarray_linalg::vector::outer;
33

44
#[test]
5-
fn outer_() {
5+
fn outer_3x4() {
66
let dist = RealNormal::<f64>::new(0.0, 1.0);
7-
let m = 2;
7+
let m = 3;
8+
let n = 4;
9+
let a = Array::random(m, dist);
10+
let b = Array::random(n, dist);
11+
println!("a = \n{:?}", &a);
12+
println!("b = \n{:?}", &b);
13+
let ab = outer(&a, &b);
14+
println!("ab = \n{:?}", &ab);
15+
for i in 0..m {
16+
for j in 0..n {
17+
ab[(i, j)].assert_close(a[i] * b[j], 1e-7);
18+
}
19+
}
20+
}
21+
22+
#[test]
23+
fn outer_4x3() {
24+
let dist = RealNormal::<f64>::new(0.0, 1.0);
25+
let m = 4;
826
let n = 3;
927
let a = Array::random(m, dist);
1028
let b = Array::random(n, dist);

0 commit comments

Comments
 (0)