add FEs and tests #2
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Comprehensive Tests | |
| on: | |
| push: | |
| branches: [ "*" ] | |
| pull_request: | |
| branches: [ "*" ] | |
| schedule: | |
| # Run tests weekly on Sundays at 2 AM UTC | |
| - cron: '0 2 * * 0' | |
| jobs: | |
| test-matrix: | |
| name: Test Python ${{ matrix.python-version }} on ${{ matrix.os }} | |
| runs-on: ${{ matrix.os }} | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| os: [ubuntu-latest, macos-latest] | |
| python-version: ["3.9", "3.10", "3.11", "3.12"] | |
| exclude: | |
| # Skip some combinations to reduce CI time | |
| - os: macos-latest | |
| python-version: "3.9" | |
| - os: macos-latest | |
| python-version: "3.10" | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v4 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Cache pip dependencies | |
| uses: actions/cache@v3 | |
| with: | |
| path: ~/.cache/pip | |
| key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ hashFiles('pyproject.toml') }} | |
| restore-keys: | | |
| ${{ runner.os }}-pip-${{ matrix.python-version }}- | |
| ${{ runner.os }}-pip- | |
| - name: Install system dependencies (Ubuntu) | |
| if: matrix.os == 'ubuntu-latest' | |
| run: | | |
| sudo apt-get update | |
| sudo apt-get install -y build-essential | |
| - name: Install Python dependencies | |
| run: | | |
| python -m pip install --upgrade pip wheel setuptools | |
| pip install pytest pytest-cov pytest-xdist | |
| - name: Install jaxonometrics | |
| run: | | |
| pip install -e . | |
| - name: Install test dependencies | |
| run: | | |
| pip install pyfixest pandas | |
| - name: Verify installation | |
| run: | | |
| python -c "import jaxonometrics; print('jaxonometrics version:', jaxonometrics.__version__)" | |
| python -c "import jax; print('JAX version:', jax.__version__)" | |
| python -c "import pyfixest; print('pyfixest available')" | |
| - name: Run basic tests | |
| run: | | |
| python -m pytest tests/test_linear.py -v | |
| - name: Run fixed effects tests | |
| run: | | |
| python -m pytest tests/test_fe.py -v | |
| - name: Run all tests with coverage | |
| run: | | |
| python -m pytest tests/ -v --cov=jaxonometrics --cov-report=xml --cov-report=term | |
| - name: Upload coverage to Codecov | |
| if: matrix.python-version == '3.11' && matrix.os == 'ubuntu-latest' | |
| uses: codecov/codecov-action@v3 | |
| with: | |
| file: ./coverage.xml | |
| flags: unittests | |
| name: codecov-umbrella | |
| test-performance: | |
| name: Performance Tests | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v4 | |
| - name: Set up Python 3.11 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: "3.11" | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install pytest pytest-benchmark | |
| pip install -e . | |
| pip install pyfixest pandas | |
| - name: Run performance benchmarks | |
| run: | | |
| python -c " | |
| import numpy as np | |
| import time | |
| import jax.numpy as jnp | |
| from jaxonometrics import LinearRegression | |
| # Performance test | |
| print('Running performance test...') | |
| np.random.seed(42) | |
| n_obs = 10000 | |
| X = jnp.asarray(np.random.randn(n_obs, 5)) | |
| y = jnp.asarray(np.random.randn(n_obs)) | |
| model = LinearRegression(solver='lineax') | |
| start_time = time.time() | |
| model.fit(X, y) | |
| end_time = time.time() | |
| print(f'Linear regression on {n_obs} obs took {end_time - start_time:.4f} seconds') | |
| print('Performance test passed!') | |
| " | |
| test-examples: | |
| name: Test Examples and Documentation | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v4 | |
| - name: Set up Python 3.11 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: "3.11" | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install jupyter nbconvert | |
| pip install -e . | |
| pip install pyfixest pandas matplotlib seaborn | |
| - name: Test example notebooks (if any) | |
| run: | | |
| # Test any Jupyter notebooks in nb/ directory | |
| if [ -d "nb" ]; then | |
| echo "Testing notebooks..." | |
| for notebook in nb/*.ipynb; do | |
| if [ -f "$notebook" ]; then | |
| echo "Testing $notebook" | |
| jupyter nbconvert --to script --execute "$notebook" | |
| fi | |
| done | |
| else | |
| echo "No notebooks directory found, skipping notebook tests" | |
| fi | |
| - name: Test README examples | |
| run: | | |
| python -c " | |
| # Test basic API examples from README | |
| import numpy as np | |
| import jax.numpy as jnp | |
| from jaxonometrics import LinearRegression | |
| print('Testing basic LinearRegression API...') | |
| # Generate sample data | |
| np.random.seed(42) | |
| X = jnp.asarray(np.random.randn(100, 3)) | |
| y = jnp.asarray(np.random.randn(100)) | |
| # Test basic fitting | |
| model = LinearRegression(solver='lineax') | |
| model.fit(X, y) | |
| # Test prediction | |
| y_pred = model.predict(X) | |
| print(f'Coefficients shape: {model.params[\"coef\"].shape}') | |
| print(f'Predictions shape: {y_pred.shape}') | |
| print('README examples test passed!') | |
| " |