r2rs-nmath 0.1.1

Statistics programming for Rust based on R's nmath package
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

// "Whatever you do, work at it with all your heart, as working for the Lord,
// not for human masters, since you know that you will receive an inheritance
// from the Lord as a reward. It is the Lord Christ you are serving."
// (Col 3:23-24)

use strafe_testing::{
    r::{RString, RTester},
    r_assert_relative_equal_result,
};

use crate::{distribution::tukey::TukeyBuilder, traits::Distribution};

pub fn probability_inner(q: f64, means: f64, df: f64, ranges: f64, lower_tail: bool) {
    let mut builder = TukeyBuilder::new();
    builder.with_means(means);
    builder.with_df(df);
    builder.with_ranges(ranges);
    let unif = builder.build();

    let rust_ans = unif.probability(q, lower_tail);
    let r_ans = RTester::new()
        .set_display(&RString::from_string(format!(
            "ptukey({}, {}, {}, {}, lower.tail={}, log={})",
            RString::from_f64(q),
            RString::from_f64(means),
            RString::from_f64(df),
            RString::from_f64(ranges),
            RString::from_bool(lower_tail),
            RString::from_bool(false),
        )))
        .run()
        .expect("R running error")
        .as_f64();

    r_assert_relative_equal_result!(r_ans, &rust_ans);
}

pub fn log_probability_inner(q: f64, ranges: f64, means: f64, df: f64, lower_tail: bool) {
    let mut builder = TukeyBuilder::new();
    builder.with_means(means);
    builder.with_df(df);
    builder.with_ranges(ranges);
    let unif = builder.build();

    let rust_ans = unif.log_probability(q, lower_tail);
    let r_ans = RTester::new()
        .set_display(&RString::from_string(format!(
            "ptukey({}, {}, {}, {}, lower.tail={}, log={})",
            RString::from_f64(q),
            RString::from_f64(means),
            RString::from_f64(df),
            RString::from_f64(ranges),
            RString::from_bool(lower_tail),
            RString::from_bool(true),
        )))
        .run()
        .expect("R running error")
        .as_f64();

    r_assert_relative_equal_result!(r_ans, &rust_ans);
}

pub fn quantile_inner(p: f64, ranges: f64, means: f64, df: f64, lower_tail: bool) {
    let mut builder = TukeyBuilder::new();
    builder.with_means(means);
    builder.with_df(df);
    builder.with_ranges(ranges);
    let unif = builder.build();

    let rust_ans = unif.quantile(p, lower_tail);
    let r_ans = RTester::new()
        .set_display(&RString::from_string(format!(
            "qtukey({}, {}, {}, {}, lower.tail={}, log={})",
            RString::from_f64(p),
            RString::from_f64(means),
            RString::from_f64(df),
            RString::from_f64(ranges),
            RString::from_bool(lower_tail),
            RString::from_bool(false),
        )))
        .run()
        .expect("R running error")
        .as_f64();

    r_assert_relative_equal_result!(r_ans, &rust_ans);
}

pub fn log_quantile_inner(p: f64, ranges: f64, means: f64, df: f64, lower_tail: bool) {
    let mut builder = TukeyBuilder::new();
    builder.with_means(means);
    builder.with_df(df);
    builder.with_ranges(ranges);
    let unif = builder.build();

    let rust_ans = unif.log_quantile(p, lower_tail);
    let r_ans = RTester::new()
        .set_display(&RString::from_string(format!(
            "qtukey({}, {}, {}, {}, lower.tail={}, log={})",
            RString::from_f64(p),
            RString::from_f64(means),
            RString::from_f64(df),
            RString::from_f64(ranges),
            RString::from_bool(lower_tail),
            RString::from_bool(true),
        )))
        .run()
        .expect("R running error")
        .as_f64();

    r_assert_relative_equal_result!(r_ans, &rust_ans);
}