hypors 0.3.0

Hypothesis Testing with Rust
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

#[cfg(test)]
mod tests_proportion {
    use hypors::common::TailType;
    use hypors::proportion::{prop_sample_size, z_test, z_test_ind};

    const EPSILON: f64 = 0.001; // Tolerance for floating-point comparisons

    #[test]
    fn test_z_test() {
        let data = vec![1, 1, 1, 0, 0];
        let null_prop = 0.5;
        let alpha = 0.05;

        let result = z_test(data, null_prop, TailType::Two, alpha).unwrap();

        let expected_z_statistic = 0.447;
        let expected_p_value = 0.655;
        let expected_null_hypothesis = "H0: p = 0.5";
        let expected_alt_hypothesis = "Ha: p ≠ 0.5";

        assert!((result.test_statistic - expected_z_statistic).abs() < EPSILON);
        assert!((result.p_value - expected_p_value).abs() < EPSILON);

        assert_eq!(result.null_hypothesis, expected_null_hypothesis);
        assert_eq!(result.alt_hypothesis, expected_alt_hypothesis);

        assert_eq!(result.reject_null, false);
    }

    #[test]
    fn test_z_test_ind_unpooled() {
        let data1 = vec![1, 1, 1, 0, 0];
        let data2 = vec![1, 1, 0, 0, 0];
        let alpha = 0.05;

        let result = z_test_ind(data1, data2, TailType::Two, alpha, false).unwrap();

        let expected_z_statistic = 0.645;
        let expected_p_value = 0.518;
        let expected_null_hypothesis = "H0: p1 = p2";
        let expected_alt_hypothesis = "Ha: p1 ≠ p2";

        assert!((result.test_statistic - expected_z_statistic).abs() < EPSILON);
        assert!((result.p_value - expected_p_value).abs() < EPSILON);

        assert_eq!(result.null_hypothesis, expected_null_hypothesis);
        assert_eq!(result.alt_hypothesis, expected_alt_hypothesis);

        assert_eq!(result.reject_null, false);
    }

    #[test]
    fn test_z_test_ind_pooled() {
        let data1 = vec![1, 1, 1, 0, 0];
        let data2 = vec![1, 1, 0, 0, 0];
        let alpha = 0.05;

        let result = z_test_ind(data1, data2, TailType::Two, alpha, true).unwrap();

        let expected_z_statistic = 0.632;
        let expected_p_value = 0.527;
        let expected_null_hypothesis = "H0: p1 = p2";
        let expected_alt_hypothesis = "Ha: p1 ≠ p2";

        assert!((result.test_statistic - expected_z_statistic).abs() < 0.001);
        assert!((result.p_value - expected_p_value).abs() < 0.001);

        assert_eq!(result.null_hypothesis, expected_null_hypothesis);
        assert_eq!(result.alt_hypothesis, expected_alt_hypothesis);

        assert_eq!(result.reject_null, false);
    }

    #[test]
    fn test_prop_sample_size() {
        let p1 = 0.4;
        let p2 = 0.6;
        let alpha = 0.05;
        let power = 0.80;

        let n = prop_sample_size(p1, p2, alpha, power);
        let expected_sample_size = 97.0;

        assert!((n - expected_sample_size).abs() < 1.0);
    }
}