so-python 0.3.0

High-performance statistical computing library written in Rust, exposed to Python via PyO3
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

#!/usr/bin/env python3
"""Test script to verify statoxide Python module structure"""

import sys
import os

# Add the build directory to path for testing
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

try:
    # Test different import styles
    print("Testing statoxide Python module imports...")
    
    # Test 1: Direct import with alias
    import statoxide as so
    print("✓ Successfully imported statoxide as 'so'")
    print(f"  Version: {so.version()}")
    
    # Test 2: Import core classes
    from statoxide import DataFrame, Series, Formula
    print("✓ Successfully imported DataFrame, Series, Formula from top level")
    
    # Create test data
    df = DataFrame({
        "x": [1.0, 2.0, 3.0, 4.0, 5.0],
        "y": [2.0, 4.0, 6.0, 8.0, 10.0],
        "group": [1.0, 1.0, 2.0, 2.0, 2.0]
    })
    print(f"✓ Created DataFrame: {df}")
    
    # Test 3: Access submodules
    print("\nTesting submodule access:")
    
    # Stats submodule
    mean_x = so.stats.mean([1.0, 2.0, 3.0, 4.0, 5.0])
    print(f"  so.stats.mean: {mean_x}")
    
    # Models submodule
    print(f"  so.models module available")
    
    # TSA submodule  
    print(f"  so.tsa module available")
    
    # Utils submodule
    train, test = so.utils.train_test_split([1.0, 2.0, 3.0, 4.0, 5.0], 0.2)
    print(f"  so.utils.train_test_split: train={len(train)}, test={len(test)}")
    
    # Test 4: Top-level functions
    print("\nTesting top-level functions:")
    mean_val = so.mean([1.0, 2.0, 3.0])
    print(f"  so.mean: {mean_val}")
    
    corr = so.correlation([1.0, 2.0, 3.0], [2.0, 4.0, 6.0])
    print(f"  so.correlation: {corr}")
    
    # Test 5: Mixed effects model (placeholder)
    print("\nTesting mixed effects model (placeholder):")
    result = so.models.mixed_effects(df, "y ~ x + (1 | group)")
    print(f"  Mixed effects result keys: {list(result.keys())}")
    
    print("\n✅ All imports successful!")
    print("\nAvailable import styles:")
    print("1. import statoxide as so")
    print("2. from statoxide import DataFrame, Series, Formula")
    print("3. Access submodules via so.stats, so.models, etc.")
    print("4. Direct function calls: so.mean(), so.correlation()")
    
except ImportError as e:
    print(f"❌ Import error: {e}")
    print("\nNote: You need to build the module first with:")
    print("  cd crates/so-python")
    print("  maturin develop")
    sys.exit(1)
except Exception as e:
    print(f"❌ Error: {e}")
    import traceback
    traceback.print_exc()
    sys.exit(1)