diff --git a/hercules_samples/dot/src/main.rs b/hercules_samples/dot/src/main.rs index 3d282ae2568a778edd70fc8cb077ac509f29536e..6d4cf3800563cd17b48eb744d2e30ffe8c641bf5 100644 --- a/hercules_samples/dot/src/main.rs +++ b/hercules_samples/dot/src/main.rs @@ -9,6 +9,12 @@ fn main() { let mut a = vec![0.0, 1.0, 0.0, 2.0, 0.0, 3.0, 0.0, 4.0]; let mut b = vec![0.0, 5.0, 0.0, 6.0, 0.0, 7.0, 0.0, 8.0]; let c = unsafe { dot(a.as_mut_ptr(), b.as_mut_ptr(), 8).await }; - println!("{}", c,); + println!("{}", c); + assert_eq!(c, 70.0); }); } + +#[test] +fn dot_test() { + main(); +} diff --git a/hercules_samples/fac/src/main.rs b/hercules_samples/fac/src/main.rs index e29860ee6942b8cbea3d084d4de0310f7fb6cf2a..e3a307fcfd521217f2956e77f25332a54e6befee 100644 --- a/hercules_samples/fac/src/main.rs +++ b/hercules_samples/fac/src/main.rs @@ -8,5 +8,11 @@ fn main() { async_std::task::block_on(async { let f = unsafe { fac(8).await }; println!("{}", f); + assert_eq!(f, 40320); }); } + +#[test] +fn fac_test() { + main(); +} diff --git a/hercules_samples/matmul/src/main.rs b/hercules_samples/matmul/src/main.rs index 26168d4b55c270b50957b0dc5a8792dbff65c5dc..f3ceba93d88402443b503a232abd5af71eb9ca58 100644 --- a/hercules_samples/matmul/src/main.rs +++ b/hercules_samples/matmul/src/main.rs @@ -15,5 +15,14 @@ fn main() { matmul(a.as_mut_ptr(), b.as_mut_ptr(), c.as_mut_ptr(), 2, 2, 2).await; } println!("[[{}, {}], [{}, {}]]", c[0], c[1], c[2], c[3]); + assert_eq!(c[0], 19.0); + assert_eq!(c[1], 22.0); + assert_eq!(c[2], 43.0); + assert_eq!(c[3], 50.0); }); } + +#[test] +fn matmul_test() { + main(); +}