"""
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()