Build out Rust unit tests/coverage reporting

.flox/env/manifest.toml

@@ -7,7 +7,7 @@ pulumi-python.pkg-path = "pulumiPackages.pulumi-python"
 ruff.pkg-path = "ruff"
 ruff.version = "0.14.7"
 uv.pkg-path = "uv"
-vulture.pkg-path = "python313Packages.vulture"
+vulture.pkg-path = "python312Packages.vulture"
 yamllint.pkg-path = "yamllint"
 nushell.pkg-path = "nushell"
 fselect.pkg-path = "fselect"
@@ -17,7 +17,8 @@ mask.pkg-path = "mask"
 bacon.pkg-path = "bacon"
 cargo-watch.pkg-path = "cargo-watch"
 cargo-nextest.pkg-path = "cargo-nextest"
-cargo-tarpaulin.pkg-path = "cargo-tarpaulin"
+cargo-llvm-cov.pkg-path = "cargo-llvm-cov"
+cargo-llvm-cov.pkg-group = "rust-toolchain"
 cargo.pkg-path = "cargo"
 cargo.pkg-group = "rust-toolchain"
 rustc.pkg-path = "rustc"
@@ -28,6 +29,8 @@ rustfmt.pkg-path = "rustfmt"
 rustfmt.pkg-group = "rust-toolchain"
 rust-lib-src.pkg-path = "rustPlatform.rustLibSrc"
 rust-lib-src.pkg-group = "rust-toolchain"
+llvm.pkg-path = "llvm"
+llvm.pkg-group = "rust-toolchain"
 libiconv.pkg-path = "libiconv"
 libiconv.systems = ["aarch64-darwin", "x86_64-darwin"]
 rust-analyzer.pkg-path = "rust-analyzer"
@@ -42,6 +45,7 @@ pkgconf.pkg-path = "pkgconf"
 direnv.pkg-path = "direnv"
 jq.pkg-path = "jq"
 markdownlint-cli.pkg-path = "markdownlint-cli"
+pre-commit.pkg-path = "pre-commit"
 
 [hook]
 on-activate = '''

.github/workflows/launch_infrastructure.yaml

@@ -49,16 +49,12 @@ jobs:
         with:
           role-to-assume: ${{ secrets.AWS_IAM_INFRASTRUCTURE_ROLE_ARN }}
           aws-region: ${{ secrets.AWS_REGION }}
+      - name: Set up Docker Buildx
+        if: steps.changes.outputs.service == 'true' || github.event_name == 'schedule'
+        uses: docker/setup-buildx-action@v3
       - name: Install Flox
         if: steps.changes.outputs.service == 'true' || github.event_name == 'schedule'
         uses: flox/install-flox-action@v2
-      - name: Conditionally download artifacts
-        if: (steps.changes.outputs.service == 'true' || github.event_name == 'schedule') && matrix.service == 'equitypricemodel'
-        uses: flox/activate-action@v1
-        env:
-          AWS_S3_ARTIFACTS_BUCKET_NAME: ${{ secrets.AWS_S3_ARTIFACTS_BUCKET_NAME }}
-        with:
-          command: mask models artifacts download equitypricemodel
       - name: Build ${{ matrix.service }} image
         if: steps.changes.outputs.service == 'true' || github.event_name == 'schedule'
         uses: flox/activate-action@v1

.github/workflows/run_code_checks.yaml

@@ -0,0 +1,93 @@
+---
+name: Code checks
+run-name: Code checks
+on:
+  pull_request:
+jobs:
+  run_rust_code_checks:
+    name: Run Rust code checks
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+      - name: Free disk space
+        run: |
+          echo "Disk space before cleanup:"
+          df -h
+          sudo rm -rf /usr/share/dotnet
+          sudo rm -rf /opt/ghc
+          sudo rm -rf /usr/local/share/boost
+          sudo rm -rf "$AGENT_TOOLSDIRECTORY"
+          echo "Disk space after cleanup:"
+          df -h
+      - name: Install Flox
+        uses: flox/install-flox-action@v2
+      - name: Cache Rust dependencies
+        uses: Swatinem/rust-cache@v2
+        with:
+          workspaces: .
+          cache-on-failure: false
+          shared-key: rust-continuous-integration
+          save-if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}
+      - name: Run Rust code checks
+        uses: flox/activate-action@v1
+        with:
+          command: mask development rust all
+      - name: Upload test coverage results
+        uses: actions/upload-artifact@v4
+        with:
+          name: ${{ github.job }}_test_coverage
+          path: .coverage_output/rust.xml
+          if-no-files-found: error
+          overwrite: true
+  run_python_code_checks:
+    name: Run Python code checks
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+      - name: Install Flox
+        uses: flox/install-flox-action@v2
+      - name: Run Python code checks
+        uses: flox/activate-action@v1
+        with:
+          command: mask development python all
+      - name: Upload test coverage results
+        uses: actions/upload-artifact@v4
+        with:
+          name: ${{ github.job }}_test_coverage
+          path: .coverage_output/python.xml
+          if-no-files-found: error
+          overwrite: true
+  run_markdown_code_checks:
+    name: Run Markdown code checks
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+      - name: Install Flox
+        uses: flox/install-flox-action@v2
+      - name: Run Markdown code checks
+        uses: flox/activate-action@v1
+        with:
+          command: mask development markdown all
+  upload_test_coverage:
+    needs:
+      - run_rust_code_checks
+      - run_python_code_checks
+    if: ${{ always() && (needs.run_rust_code_checks.result == 'success' || needs.run_python_code_checks.result == 'success') }}
+    name: Upload coverage to Coveralls
+    runs-on: ubuntu-latest
+    steps:
+      - name: Download test coverage results
+        uses: actions/download-artifact@v4
+        with:
+          pattern: '*_test_coverage'
+          path: .coverage_output/
+          merge-multiple: true
+          run-id: ${{ github.run_id }}
+      - name: Upload coverage to Coveralls
+        uses: coverallsapp/github-action@v2
+        with:
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          files: .coverage_output/python.xml .coverage_output/rust.xml

.gitignore

@@ -9,8 +9,8 @@
 __pycache__/
 .envrc
 .env.nu
-coverage/
-.coverage
+.coverage_output/
+.coverage*
 *.csv
 *.egg-info/
 wandb/
@@ -22,3 +22,4 @@ data/
 **/*.json
 .claude/tasks/
 .scratchpad/
+

CLAUDE.md

@@ -31,7 +31,7 @@ This is a collection of guidelines and references.
 - Use Polars for [Python](https://docs.pola.rs/api/python/stable/reference/index.html) and
   [Rust](https://docs.rs/polars/latest/polars/) dataframes
 - Use `typing` module `cast` function for `tinygrad` method outputs when necessary with union types
-- Write `pytest` functions for Python tests
+- Write `pytest` functions for Python tests using plain functions, not class-based test organization
 - Ensure Rust and Python automated test suites achieve at least 90% line or statement coverage per service or library
 - Exclude generated code, third-party code, tooling boilerplate, and anything explicitly excluded in this repository
   from test coverage calculations