Skip to content

Commit 1a0c602

Browse files
committed
Use new asserts in tests
1 parent c016bc5 commit 1a0c602

File tree

7 files changed

+19
-17
lines changed

7 files changed

+19
-17
lines changed

tests/det.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ mod $modname {
1111
let (e, _) = a.$clone().eigh().unwrap();
1212
let deth = a.$clone().deth().unwrap();
1313
let det_eig = e.iter().fold(1.0, |x, y| x * y);
14-
deth.assert_close(det_eig, 1.0e-7);
14+
assert_rclose!(deth, det_eig, 1.0e-7);
1515
}
1616
}
1717
}} // impl_test

tests/eigh.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ mod $modname {
2222
let (e, vecs) = a.$clone().eigh().unwrap();
2323
let s = vecs.t().dot(&a).dot(&vecs);
2424
for i in 0..3 {
25-
e[i].assert_close(s[(i, i)], 1e-7);
25+
assert_rclose!(e[i], s[(i, i)], 1e-7);
2626
}
2727
}
2828
}

tests/header.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
extern crate rand_extra;
33
extern crate ndarray;
44
extern crate ndarray_rand;
5+
#[macro_use]
6+
#[allow(unused_imports)]
57
extern crate ndarray_linalg;
68
extern crate num_traits;
79

tests/opnorm.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@ macro_rules! impl_test {
55
#[test]
66
fn $funcname() {
77
let a = $a;
8-
a.opnorm_1().assert_close($op1, 1e-7);
9-
a.opnorm_i().assert_close($opi, 1e-7);
10-
a.opnorm_f().assert_close($opf, 1e-7);
8+
assert_rclose!(a.opnorm_1(), $op1, 1e-7);
9+
assert_rclose!(a.opnorm_i(), $opi, 1e-7);
10+
assert_rclose!(a.opnorm_f(), $opf, 1e-7);
1111
}
1212
}} // impl_test
1313

tests/outer.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ fn outer_3x4() {
1414
println!("ab = \n{:?}", &ab);
1515
for i in 0..m {
1616
for j in 0..n {
17-
ab[(i, j)].assert_close(a[i] * b[j], 1e-7);
17+
assert_rclose!(ab[(i, j)], a[i] * b[j], 1e-7);
1818
}
1919
}
2020
}
@@ -32,7 +32,7 @@ fn outer_4x3() {
3232
println!("ab = \n{:?}", &ab);
3333
for i in 0..m {
3434
for j in 0..n {
35-
ab[(i, j)].assert_close(a[i] * b[j], 1e-7);
35+
assert_rclose!(ab[(i, j)], a[i] * b[j], 1e-7);
3636
}
3737
}
3838
}

tests/trace.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,5 @@ include!("header.rs");
44
fn trace() {
55
let r_dist = RealNormal::new(0., 1.);
66
let a = Array::<f64, _>::random((3, 3), r_dist);
7-
a.trace().unwrap().assert_close(a[(0, 0)] + a[(1, 1)] + a[(2, 2)], 1e-7);
7+
assert_rclose!(a.trace().unwrap(), a[(0, 0)] + a[(1, 1)] + a[(2, 2)], 1e-7);
88
}

tests/vector.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,37 +3,37 @@ include!("header.rs");
33
#[test]
44
fn vector_norm() {
55
let a = Array::range(1., 10., 1.);
6-
a.norm().assert_close(285.0.sqrt(), 1e-7);
6+
assert_rclose!(a.norm(), 285.0.sqrt(), 1e-7);
77
}
88

99
#[test]
1010
fn vector_norm_l1() {
1111
let a = arr1(&[1.0, -1.0]);
12-
a.norm_l1().assert_close(2.0, 1e-7);
12+
assert_rclose!(a.norm_l1(), 2.0, 1e-7);
1313
let b = arr2(&[[0.0, -1.0], [1.0, 0.0]]);
14-
b.norm_l1().assert_close(2.0, 1e-7);
14+
assert_rclose!(b.norm_l1(), 2.0, 1e-7);
1515
}
1616

1717
#[test]
1818
fn vector_norm_max() {
1919
let a = arr1(&[1.0, 1.0, -3.0]);
20-
a.norm_max().assert_close(3.0, 1e-7);
20+
assert_rclose!(a.norm_max(), 3.0, 1e-7);
2121
let b = arr2(&[[1.0, 3.0], [1.0, -4.0]]);
22-
b.norm_max().assert_close(4.0, 1e-7);
22+
assert_rclose!(b.norm_max(), 4.0, 1e-7);
2323
}
2424

2525
#[test]
2626
fn vector_norm_l1_rc() {
2727
let a = rcarr1(&[1.0, -1.0]);
28-
a.norm_l1().assert_close(2.0, 1e-7);
28+
assert_rclose!(a.norm_l1(), 2.0, 1e-7);
2929
let b = rcarr2(&[[0.0, -1.0], [1.0, 0.0]]);
30-
b.norm_l1().assert_close(2.0, 1e-7);
30+
assert_rclose!(b.norm_l1(), 2.0, 1e-7);
3131
}
3232

3333
#[test]
3434
fn vector_norm_max_rc() {
3535
let a = rcarr1(&[1.0, 1.0, -3.0]);
36-
a.norm_max().assert_close(3.0, 1e-7);
36+
assert_rclose!(a.norm_max(), 3.0, 1e-7);
3737
let b = rcarr2(&[[1.0, 3.0], [1.0, -4.0]]);
38-
b.norm_max().assert_close(4.0, 1e-7);
38+
assert_rclose!(b.norm_max(), 4.0, 1e-7);
3939
}

0 commit comments

Comments
 (0)