Skip to content

Commit 64af5a9

Browse files
committed
Basic tests for scalar array arith API calls
1 parent a4aee8a commit 64af5a9

File tree

2 files changed

+19
-0
lines changed

2 files changed

+19
-0
lines changed
File renamed without changes.

tests/scalar_arith.rs

+19
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
use ::arrayfire::*;
2+
3+
#[allow(non_snake_case)]
4+
#[test]
5+
fn check_scalar_arith() {
6+
let dims = Dim4::new(&[5, 5, 1, 1]);
7+
let A = randu::<f32>(dims);
8+
let s: f32 = 2.0;
9+
let scalar_as_lhs = s * &A;
10+
let scalar_as_rhs = &A * s;
11+
let C = constant(s, dims);
12+
let no_scalars = A * C;
13+
let scalar_res_comp = eq(&scalar_as_lhs, &scalar_as_rhs, false);
14+
let res_comp = eq(&scalar_as_lhs, &no_scalars, false);
15+
let scalar_res = all_true_all(&scalar_res_comp);
16+
let res = all_true_all(&res_comp);
17+
18+
assert_eq!(scalar_res.0, res.0);
19+
}

0 commit comments

Comments
 (0)