Enable test suite (#3)

Author: dirmeier

.github/workflows/ci.yaml

@@ -20,7 +20,7 @@ jobs:
       - precommit
     strategy:
       matrix:
-        python-version: [ 3.11 ]
+        python-version: [ 3.12 ]
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}
@@ -36,3 +36,30 @@ jobs:
       - name: Run lints
         run: |
           make lints
+
+  tests:
+    runs-on: ubuntu-latest
+    needs:
+      - lints
+    strategy:
+      matrix:
+        python-version: [ 3.12 ]
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v3
+        with:
+          python-version: ${{ matrix.python-version }}
+      - uses: astral-sh/setup-uv@v5
+        with:
+          version: "latest"
+      - name: Install dependencies
+        run: |
+          uv sync --dev
+      - name: Run tests
+        run: |
+          make tests
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v5
+        env:
+          CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}

blaxbird/__init__.py

@@ -1,6 +1,6 @@
 """blaxbird: A high-level API for building and training Flax NNX models."""
 
-__version__ = "0.0.2"
+__version__ = "0.1.0"
 
 from blaxbird._src.checkpointer import get_default_checkpointer
 from blaxbird._src.trainer import train_fn

blaxbird/_src/test_blaxbird.py

@@ -0,0 +1,5 @@
+import chex
+
+
+def test_blaxbird():
+  chex.assert_equal(1, 1)