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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

#[cfg(test)]
mod tests_z_test {
    use hypors::common::TailType;
    use hypors::z::{z_sample_size, z_test, z_test_ind, z_test_paired};

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

    #[test]
    fn test_z_test() {
        let data = vec![2.0, 3.0, 5.0, 7.0, 11.0];
        let pop_mean = 5.0;
        let pop_std = 2.0;
        let alpha = 0.05;

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

        let expected_z_statistic = 0.670;
        let expected_p_value = 0.502;
        let expected_ci_lower = 3.846954;
        let expected_ci_upper = 7.353045;
        let expected_null_hypothesis = "H0: µ = 5";
        let expected_alt_hypothesis = "Ha: µ ≠ 5";

        assert!((result.test_statistic - expected_z_statistic).abs() < EPSILON);
        assert!((result.p_value - expected_p_value).abs() < EPSILON);
        assert_eq!(result.reject_null, false);

        assert!((result.confidence_interval.0 - expected_ci_lower).abs() < EPSILON);
        assert!((result.confidence_interval.1 - expected_ci_upper).abs() < EPSILON);

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

    #[test]
    fn test_z_test_paired() {
        let data1 = vec![2.0, 3.0, 5.0, 7.0, 11.0];
        let data2 = vec![1.0, 3.0, 6.0, 7.0, 10.0];
        let pop_std_diff = 1.5;
        let alpha = 0.05;

        let result = z_test_paired(data1, data2, pop_std_diff, TailType::Two, alpha).unwrap();

        let expected_z_statistic = 0.298;
        let expected_p_value = 0.765;
        let expected_ci_lower = -1.114783;
        let expected_ci_upper = 1.514783;
        let expected_null_hypothesis = "H0: µ1 = µ2";
        let expected_alt_hypothesis = "Ha: µ1 ≠ µ2";

        assert!((result.test_statistic - expected_z_statistic).abs() < EPSILON);
        assert!((result.p_value - expected_p_value).abs() < EPSILON);
        assert_eq!(result.reject_null, false);

        println!("{} {}", result.confidence_interval.0, expected_ci_lower);
        println!("{} {}", result.confidence_interval.1, expected_ci_upper);

        assert!((result.confidence_interval.0 - expected_ci_lower).abs() < EPSILON);
        assert!((result.confidence_interval.1 - expected_ci_upper).abs() < EPSILON);

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

    #[test]
    fn test_z_test_ind() {
        let data1 = vec![2.0, 3.0, 5.0, 7.0, 11.0];
        let data2 = vec![1.0, 3.0, 6.0, 7.0, 10.0];
        let pop_std1 = 2.0;
        let pop_std2 = 1.5;
        let alpha = 0.05;

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

        let expected_z_statistic = 0.179;
        let expected_p_value = 0.858;
        let expected_ci_lower = -1.991306;
        let expected_ci_upper = 2.391306;
        let expected_null_hypothesis = "H0: µ1 = µ2";
        let expected_alt_hypothesis = "Ha: µ1 ≠ µ2";

        assert!((result.test_statistic - expected_z_statistic).abs() < EPSILON);
        assert!((result.p_value - expected_p_value).abs() < EPSILON);
        assert_eq!(result.reject_null, false);

        assert!((result.confidence_interval.0 - expected_ci_lower).abs() < EPSILON);
        assert!((result.confidence_interval.1 - expected_ci_upper).abs() < EPSILON);

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

    #[test]
    fn test_z_sample_size() {
        let effect_size = 0.3;
        let alpha = 0.05;
        let power = 0.80;
        let std_dev = 1.0;
        let tail = TailType::Two;

        let n = z_sample_size(effect_size, alpha, power, std_dev, tail);

        let expected_sample_size = 87.79;

        assert!(
            (n - expected_sample_size).abs() < 1.0,
            "Sample size is incorrect"
        );
    }
}