la-stack 0.5.2

Fast, stack-allocated linear algebra for fixed dimensions
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

#![forbid(unsafe_code)]

//! Determinant multilinearity for arbitrary canonical polynomial entries.

#![cfg(feature = "exact")]

use std::array::from_fn;

use la_stack::prelude::*;
use pastey::paste;
use proptest::prelude::*;

#[path = "common/proptest_config.rs"]
mod proptest_config;

const BUDGET: PolynomialDeterminantBudget = PolynomialDeterminantBudget {
    max_subset_transitions: 1_024,
    max_term_products: 1_000_000,
};

fn row_scaling<const D: usize>(terms: &[(u8, i16, i16)], scale: i16) {
    let mut matrix: [[SparsePolynomial; D]; D] = from_fn(|row| {
        from_fn(|col| {
            let start = 2 * (row * D + col);
            SparsePolynomial::try_from_terms(
                terms[start..start + 2]
                    .iter()
                    .map(|&(e, n, d)| (u128::from(e), BigRational::new(n.into(), d.into()))),
            )
            .unwrap()
        })
    });
    let determinant = polynomial_determinant(&matrix, BUDGET).unwrap();
    let factor = BigRational::from_integer(scale.into());
    for entry in &mut matrix[0] {
        *entry =
            SparsePolynomial::try_from_terms(entry.terms().iter().map(|(&e, c)| (e, c * &factor)))
                .unwrap();
    }
    let expected = SparsePolynomial::try_from_terms(
        determinant.terms().iter().map(|(&e, c)| (e, c * &factor)),
    )
    .unwrap();
    assert_eq!(polynomial_determinant(&matrix, BUDGET).unwrap(), expected);
}

macro_rules! dimension_properties {
    ($d:literal) => {
        paste! {
            proptest! {
                #![proptest_config(proptest_config::with_default_cases(32))]
                #[test]
                fn [<row_scaling_ $d>](
                    terms in prop::collection::vec((0_u8..6, -4_i16..5, 1_i16..7), 2 * $d * $d),
                    scale in -3_i16..4,
                ) {
                    row_scaling::<$d>(&terms, scale);
                }
            }
        }
    };
}
dimension_properties!(2);
dimension_properties!(3);
dimension_properties!(4);
dimension_properties!(5);