Test/unit testing

.github/workflows/tests.yml

@@ -1,26 +1,31 @@
 name: Tests
 
 on:
-  pull_request:
   push:
-    branches:
-      - main
+    branches: [main, dev]
+  pull_request:
+    branches: [main]
 
 jobs:
-  non-llm-tests:
+  test:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.11", "3.12", "3.13"]
 
     steps:
       - uses: actions/checkout@v4
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v5
+        uses: astral-sh/setup-uv@v4
+        with:
+          version: "latest"
 
-      - name: Set up Python
-        run: uv python install 3.11
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
 
       - name: Install dependencies
         run: uv sync --extra dev
 
       - name: Run non-LLM tests
-        run: uv run pytest -q -m "not llm"
+        run: uv run pytest -q -m "not llm" --tb=short

pyproject.toml

@@ -8,7 +8,13 @@ authors = [
     { name = "Nhat Quang Dang", email = "dangnhatquang199@gmail.com" },
 ]
 requires-python = ">=3.10,<3.14"
-dependencies = ["crewai[tools]==1.14.4", "pyyaml>=6.0", "python-dotenv>=1.0.0"]
+
+dependencies = [
+    "crewai[tools]==1.14.4",
+    "pyyaml>=6.0",
+    "python-dotenv>=1.0.0",
+    "ollama>=0.6.2",
+]
 
 [project.optional-dependencies]
 dev = ["pytest>=8.0.0", "pytest-cov>=5.0.0"]
@@ -20,6 +26,8 @@ train = "bandai.main:train"
 replay = "bandai.main:replay"
 test = "bandai.main:test"
 run_with_trigger = "bandai.main:run_with_trigger"
+kickoff = "bandai.main:run"
+pytest_unit = "bandai.main:run_pytest"
 
 [build-system]
 requires = ["hatchling"]

src/bandai/main.py

@@ -134,6 +134,11 @@ def test() -> None:
     _run_command(["crewai", "test", *sys.argv[1:]])
 
 
+def run_pytest() -> None:
+    """Run unit tests (pytest) via the project script uv run pytest_unit."""
+    _run_command(["pytest", "tests/", "-v", *sys.argv[1:]])
+
+
 def run_with_trigger() -> None:
     """Run the BandAI pipeline from an external trigger (webhook/API)."""
     run()