"""
Hypothesis Testing Example

Author: RSK World
Website: https://rskworld.in
Email: help@rskworld.in
Phone: +91 93305 39277
"""

import numpy as np
import sys
import os

# Add parent directory to path
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from hypothesis_testing import StatisticalTests


def main():
    print("Hypothesis Testing Example")
    print("=" * 70)
    
    # Generate sample data
    np.random.seed(42)
    sample1 = np.random.normal(100, 15, 30)
    sample2 = np.random.normal(105, 15, 30)
    sample3 = np.random.normal(110, 15, 30)
    
    # Create test object
    tests = StatisticalTests()
    
    # T-test
    print("\nT-Test:")
    print("=" * 70)
    tests.t_test(sample1, sample2)
    
    # ANOVA test
    print("\n" + "=" * 70)
    print("ANOVA Test:")
    print("=" * 70)
    tests.anova_test({
        'Group 1': sample1,
        'Group 2': sample2,
        'Group 3': sample3
    })
    
    # Normality test
    print("\n" + "=" * 70)
    print("Normality Test:")
    print("=" * 70)
    tests.normality_test(sample1)
    
    # Chi-square test
    print("\n" + "=" * 70)
    print("Chi-Square Test:")
    print("=" * 70)
    observed = np.array([30, 25, 20, 15, 10])
    expected = np.array([20, 20, 20, 20, 20])
    tests.chi_square_test(observed, expected)


if __name__ == "__main__":
    main()

date,GDP,Consumption,Investment,Unemployment
2020-01-01,100.0,85.0,45.0,5.2
2020-02-01,101.2,86.1,45.5,5.1
2020-03-01,102.5,87.3,46.2,5.0
2020-04-01,103.8,88.5,47.0,4.9
2020-05-01,105.1,89.7,47.8,4.8
2020-06-01,106.4,90.9,48.5,4.7
2020-07-01,107.7,92.1,49.3,4.6
2020-08-01,109.0,93.3,50.0,4.5
2020-09-01,110.3,94.5,50.8,4.4
2020-10-01,111.6,95.7,51.5,4.3
2020-11-01,112.9,96.9,52.3,4.2
2020-12-01,114.2,98.1,53.0,4.1
2021-01-01,115.5,99.3,53.8,4.0
2021-02-01,116.8,100.5,54.5,3.9
2021-03-01,118.1,101.7,55.3,3.8
2021-04-01,119.4,102.9,56.0,3.7
2021-05-01,120.7,104.1,56.8,3.6
2021-06-01,122.0,105.3,57.5,3.5
2021-07-01,123.3,106.5,58.3,3.4
2021-08-01,124.6,107.7,59.0,3.3
2021-09-01,125.9,108.9,59.8,3.2
2021-10-01,127.2,110.1,60.5,3.1
2021-11-01,128.5,111.3,61.3,3.0
2021-12-01,129.8,112.5,62.0,2.9
2022-01-01,131.1,113.7,62.8,2.8
2022-02-01,132.4,114.9,63.5,2.7
2022-03-01,133.7,116.1,64.3,2.6
2022-04-01,135.0,117.3,65.0,2.5
2022-05-01,136.3,118.5,65.8,2.4
2022-06-01,137.6,119.7,66.5,2.3
2022-07-01,138.9,120.9,67.3,2.2
2022-08-01,140.2,122.1,68.0,2.1
2022-09-01,141.5,123.3,68.8,2.0
2022-10-01,142.8,124.5,69.5,1.9
2022-11-01,144.1,125.7,70.3,1.8
2022-12-01,145.4,126.9,71.0,1.7