|
| 1 | +use core_simd::{BitMask, Mask8, SimdMask8, SimdI8}; |
| 2 | +use core::convert::TryFrom; |
| 3 | + |
| 4 | +#[cfg(target_arch = "wasm32")] |
| 5 | +use wasm_bindgen_test::*; |
| 6 | + |
| 7 | +#[cfg(target_arch = "wasm32")] |
| 8 | +wasm_bindgen_test_configure!(run_in_browser); |
| 9 | + |
| 10 | +#[test] |
| 11 | +#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)] |
| 12 | +fn mask_format_round_trip() { |
| 13 | + let ints = SimdI8::from_array([-1, 0, 0, -1]); |
| 14 | + |
| 15 | + let simd_mask = SimdMask8::try_from(ints).unwrap(); |
| 16 | + |
| 17 | + let bitmask = BitMask::from(simd_mask); |
| 18 | + |
| 19 | + let opaque_mask = Mask8::from(bitmask); |
| 20 | + |
| 21 | + let simd_mask_returned = SimdMask8::from(opaque_mask); |
| 22 | + |
| 23 | + let ints_returned = SimdI8::from(simd_mask_returned); |
| 24 | + |
| 25 | + assert_eq!(ints_returned, ints); |
| 26 | +} |
| 27 | + |
| 28 | +macro_rules! test_mask_api { |
| 29 | + { $name:ident } => { |
| 30 | + #[allow(non_snake_case)] |
| 31 | + mod $name { |
| 32 | + #[test] |
| 33 | + #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)] |
| 34 | + fn set_and_test() { |
| 35 | + let values = [true, false, false, true, false, false, true, false]; |
| 36 | + let mut mask = core_simd::$name::<8>::splat(false); |
| 37 | + for (lane, value) in values.iter().copied().enumerate() { |
| 38 | + mask.set(lane, value); |
| 39 | + } |
| 40 | + for (lane, value) in values.iter().copied().enumerate() { |
| 41 | + assert_eq!(mask.test(lane), value); |
| 42 | + } |
| 43 | + } |
| 44 | + |
| 45 | + #[test] |
| 46 | + #[should_panic] |
| 47 | + fn set_invalid_lane() { |
| 48 | + let mut mask = core_simd::$name::<8>::splat(false); |
| 49 | + mask.set(8, true); |
| 50 | + let _ = mask; |
| 51 | + } |
| 52 | + |
| 53 | + #[test] |
| 54 | + #[should_panic] |
| 55 | + fn test_invalid_lane() { |
| 56 | + let mask = core_simd::$name::<8>::splat(false); |
| 57 | + let _ = mask.test(8); |
| 58 | + } |
| 59 | + } |
| 60 | + } |
| 61 | +} |
| 62 | + |
| 63 | +mod mask_api { |
| 64 | + test_mask_api!{ Mask8 } |
| 65 | + test_mask_api!{ SimdMask8 } |
| 66 | + test_mask_api!{ BitMask } |
| 67 | +} |
0 commit comments