nalgebra/nalgebra-sparse/tests/common/mod.rs

38 lines
1.4 KiB
Rust
Raw Normal View History

2020-12-09 21:42:31 +08:00
use proptest::strategy::Strategy;
use nalgebra_sparse::csr::CsrMatrix;
use nalgebra_sparse::proptest::{csr, csc};
use nalgebra_sparse::csc::CscMatrix;
2020-12-14 23:55:06 +08:00
use std::ops::RangeInclusive;
2020-12-09 21:42:31 +08:00
2020-07-14 00:44:40 +08:00
#[macro_export]
macro_rules! assert_panics {
($e:expr) => {{
use std::panic::{catch_unwind};
use std::stringify;
let expr_string = stringify!($e);
// Note: We cannot manipulate the panic hook here, because it is global and the test
// suite is run in parallel, which leads to race conditions in the sense
// that some regular tests that panic might not output anything anymore.
// Unfortunately this means that output is still printed to stdout if
// we run cargo test -- --nocapture. But Cargo does not forward this if the test
// binary is not run with nocapture, so it is somewhat acceptable nonetheless.
let result = catch_unwind(|| $e);
if result.is_ok() {
panic!("assert_panics!({}) failed: the expression did not panic.", expr_string);
}
}};
2020-12-09 21:42:31 +08:00
}
2020-12-14 23:55:06 +08:00
pub const PROPTEST_MATRIX_DIM: RangeInclusive<usize> = 0..=6;
pub const PROPTEST_MAX_NNZ: usize = 40;
2020-12-09 21:42:31 +08:00
pub fn csr_strategy() -> impl Strategy<Value=CsrMatrix<i32>> {
2020-12-14 23:55:06 +08:00
csr(-5 ..= 5, PROPTEST_MATRIX_DIM, PROPTEST_MATRIX_DIM, PROPTEST_MAX_NNZ)
2020-12-09 21:42:31 +08:00
}
pub fn csc_strategy() -> impl Strategy<Value=CscMatrix<i32>> {
2020-12-14 23:55:06 +08:00
csc(-5 ..= 5, PROPTEST_MATRIX_DIM, PROPTEST_MATRIX_DIM, PROPTEST_MAX_NNZ)
2020-12-09 21:42:31 +08:00
}