add ci

Author: reiniscimurs

.github/workflows/main.yml

@@ -0,0 +1,31 @@
+name: Run Tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+
+      - name: Install Poetry
+        run: |
+          curl -sSL https://install.python-poetry.org | python3 -
+          echo "$HOME/.local/bin" >> $GITHUB_PATH
+
+      - name: Install dependencies
+        run: poetry install
+
+      - name: Run tests
+        run: poetry run pytest

pyproject.toml

@@ -19,6 +19,8 @@ dependencies = [
     "scipy (>=1.15.2,<2.0.0)"
 ]
 
+[tool.pytest.ini_options]
+testpaths = ["tests"]
 
 [build-system]
 requires = ["poetry-core>=2.0.0,<3.0.0"]

tests/test_model.py

@@ -1,3 +1,5 @@
+from pathlib import Path
+
 from robot_nav.models.RCPG.RCPG import RCPG
 from robot_nav.models.TD3.TD3 import TD3
 from robot_nav.models.CNNTD3.CNNTD3 import CNNTD3
@@ -7,6 +9,7 @@
 from robot_nav.sim import SIM_ENV
 import pytest
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
 
 @pytest.mark.parametrize(
     "model, state_dim",
@@ -39,7 +42,7 @@ def test_models(model, state_dim):
         training_iterations=0,
         batch_size=0,
         buffer_size=100,
-        file_names=["test_data.yml"],
+        file_names=[PROJECT_ROOT.joinpath("tests/test_data.yml")],
     )
 
     test_model.train(
@@ -79,7 +82,7 @@ def test_max_bound_models(model, state_dim):
         training_iterations=0,
         batch_size=0,
         buffer_size=100,
-        file_names=["test_data.yml"],
+        file_names=[PROJECT_ROOT.joinpath("tests/test_data.yml")],
     )
 
     test_model.train(

tests/test_utils.py

@@ -1,9 +1,12 @@
+from pathlib import Path
+
 from robot_nav.models.SAC.SAC import SAC
 from robot_nav.models.PPO.PPO import PPO, RolloutBuffer
 from robot_nav.models.RCPG.RCPG import RCPG
 from robot_nav.utils import get_buffer, RolloutReplayBuffer, ReplayBuffer
 from robot_nav.sim import SIM_ENV
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
 
 def test_buffer():
     model = SAC(
@@ -39,7 +42,7 @@ def test_buffer():
         training_iterations=0,
         batch_size=0,
         buffer_size=100,
-        file_names=["test_data.yml"],
+        file_names=[PROJECT_ROOT.joinpath("tests/test_data.yml")],
     )
     assert prefilled_buffer.count == 100
 
@@ -78,7 +81,7 @@ def test_rollout_buffer():
         training_iterations=0,
         batch_size=0,
         buffer_size=100,
-        file_names=["test_data.yml"],
+        file_names=[PROJECT_ROOT.joinpath("tests/test_data.yml")],
     )
     assert prefilled_buffer.count == 6