user0/rust
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add tests

Browse source
This commit is contained in:
Caleb Zulawski 2021-03-08 00:48:18 -05:00
parent 875b31c33f
commit a7b82adb12
3 changed files with 140 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

18
crates/core_simd/tests/masks.rs
View file

@ -59,6 +59,24 @@ macro_rules! test_mask_api {
let mask = core_simd::$name::<8>::splat(false);
let _ = mask.test(8);
}
#[test]
fn any() {
assert!(!core_simd::$name::<8>::splat(false).any());
assert!(core_simd::$name::<8>::splat(true).any());
let mut v = core_simd::$name::<8>::splat(false);
v.set(2, true);
assert!(v.any());
}
#[test]
fn all() {
assert!(!core_simd::$name::<8>::splat(false).all());
assert!(core_simd::$name::<8>::splat(true).all());
let mut v = core_simd::$name::<8>::splat(false);
v.set(2, true);
assert!(!v.all());
}
}
}
}

121
crates/core_simd/tests/ops_macros.rs
View file

@ -136,6 +136,83 @@ macro_rules! impl_binary_checked_op_test {
};
}
#[macro_export]
macro_rules! impl_common_integer_tests {
{ $vector:ident, $scalar:ident } => {
test_helpers::test_lanes! {
fn wrapping_sum<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).wrapping_sum(),
x.iter().copied().fold(0 as $scalar, $scalar::wrapping_add),
);
Ok(())
});
}
fn wrapping_product<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).wrapping_product(),
x.iter().copied().fold(1 as $scalar, $scalar::wrapping_mul),
);
Ok(())
});
}
fn and_lanes<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).and_lanes(),
x.iter().copied().fold(-1i8 as $scalar, <$scalar as core::ops::BitAnd>::bitand),
);
Ok(())
});
}
fn or_lanes<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).or_lanes(),
x.iter().copied().fold(0 as $scalar, <$scalar as core::ops::BitOr>::bitor),
);
Ok(())
});
}
fn xor_lanes<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).xor_lanes(),
x.iter().copied().fold(0 as $scalar, <$scalar as core::ops::BitXor>::bitxor),
);
Ok(())
});
}
fn max_lane<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).max_lane(),
x.iter().copied().max().unwrap(),
);
Ok(())
});
}
fn min_lane<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
$vector::<LANES>::from_array(x).min_lane(),
x.iter().copied().min().unwrap(),
);
Ok(())
});
}
}
}
}
/// Implement tests for signed integers.
#[macro_export]
macro_rules! impl_signed_tests {
@ -144,6 +221,8 @@ macro_rules! impl_signed_tests {
type Vector<const LANES: usize> = core_simd::$vector<LANES>;
type Scalar = $scalar;
impl_common_integer_tests! { Vector, Scalar }
test_helpers::test_lanes! {
fn neg<const LANES: usize>() {
test_helpers::test_unary_elementwise(
@ -241,6 +320,8 @@ macro_rules! impl_unsigned_tests {
type Vector<const LANES: usize> = core_simd::$vector<LANES>;
type Scalar = $scalar;
impl_common_integer_tests! { Vector, Scalar }
test_helpers::test_lanes_panic! {
fn rem_zero_panic<const LANES: usize>() {
let a = Vector::<LANES>::splat(42);
@ -397,6 +478,46 @@ macro_rules! impl_float_tests {
},
).unwrap();
}
fn sum<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
Vector::<LANES>::from_array(x).sum(),
x.iter().copied().fold(0 as Scalar, <Scalar as core::ops::Add>::add),
);
Ok(())
});
}
fn product<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
Vector::<LANES>::from_array(x).product(),
x.iter().copied().fold(1. as Scalar, <Scalar as core::ops::Mul>::mul),
);
Ok(())
});
}
fn max_lane<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
Vector::<LANES>::from_array(x).max_lane(),
x.iter().copied().fold(Scalar::NAN, Scalar::max),
);
Ok(())
});
}
fn min_lane<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (
Vector::<LANES>::from_array(x).min_lane(),
x.iter().copied().fold(Scalar::NAN, Scalar::min),
);
Ok(())
});
}
}
}
}

2
crates/test_helpers/src/biteq.rs
View file

@ -95,7 +95,7 @@ impl<T: BitEq> core::fmt::Debug for BitEqWrapper<'_, T> {
#[macro_export]
macro_rules! prop_assert_biteq {
{ $a:expr, $b:expr } => {
{ $a:expr, $b:expr $(,)? } => {
{
use $crate::biteq::BitEqWrapper;
let a = $a;