petektools 0.2.2

Standalone numerics & geostatistics kernels for Rust: scattered-data gridding (minimum-curvature, IDW, nearest) and a curated numeric front-door. Pure leaf with PyO3 bindings.
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
67
68
69
70
71
72
73
74
75

//! Cross-language reproducibility parity vector.
//!
//! These golden values pin the exact draw stream of `seeded_rng(seed)` +
//! `Sampler` for a fixed seed/params. The **same** vectors are re-asserted from
//! Python in the wheel smoke test (`python/tests/test_petektools.py`), so a
//! drift on *either* side — a change to the RNG/variate path in Rust, or a
//! marshalling bug in the PyO3 bindings — fails a test rather than silently
//! diverging. Seed and params are shared between the two languages by copy;
//! keep them in lockstep if either is edited.
//!
//! Exact `f64` equality is intentional: reproducibility means bit-for-bit, and
//! the literals are Python's shortest round-tripping `repr`, which parses back
//! to the identical IEEE-754 double in Rust.

use petektools::sampling::{seeded_rng, Sampler};

const SEED: u64 = 20260703;

fn draw(sampler: Sampler, n: usize) -> Vec<f64> {
    sampler.sample_n(n, &mut seeded_rng(SEED))
}

#[test]
fn uniform_parity_vector() {
    let got = draw(Sampler::new_uniform(0.0, 1.0).unwrap(), 5);
    let want = [
        0.33443602974759323,
        0.009958944617155074,
        0.5658402900563928,
        0.4914127600800473,
        0.15156206470701195,
    ];
    assert_eq!(got, want);
}

#[test]
fn normal_parity_vector() {
    let got = draw(Sampler::new_normal(0.0, 1.0).unwrap(), 5);
    let want = [
        -0.2141390202029558,
        -0.8558773516768997,
        0.3154728438139085,
        -0.02578857523830222,
        -1.0136631546527357,
    ];
    assert_eq!(got, want);
}

#[test]
fn triangular_parity_vector() {
    let got = draw(Sampler::new_triangular(0.0, 1.0, 2.0).unwrap(), 3);
    let want = [0.8178459876377621, 0.14113075226296412, 1.0681634156746074];
    assert_eq!(got, want);
}

#[test]
fn lognormal_parity_vector() {
    let got = draw(Sampler::new_lognormal(0.0, 0.25).unwrap(), 3);
    let want = [0.9478729970572154, 0.8073731403836002, 1.0820617087922064];
    assert_eq!(got, want);
}

#[test]
fn truncated_normal_parity_vector() {
    let got = draw(
        Sampler::new_truncated_normal(0.0, 1.0, -1.0, 1.0).unwrap(),
        3,
    );
    let want = [
        -0.2872218793831341,
        -0.9722860907259094,
        0.11290855771814236,
    ];
    assert_eq!(got, want);
}