Fix ci testing

.github/workflows/workflows.yaml

@@ -1,11 +1,43 @@
-name: Python package
+name: Run tests
 
 on: [push]
+
+env:
+  pytest_cpus: 2 # How many cpu cores will be used to run tests
+  pytest_cov_dir: 'dgmr' # The directory that pytest-cov will look at the coverage
+
 jobs:
-  call-run-python-tests:
-    uses: openclimatefix/.github/.github/workflows/python-test.yml@main
-    with:
-      # 0 means don't use pytest-xdist
-      pytest_numcpus: "2"
-      # pytest-cov looks at this folder
-      pytest_cov_dir: "dgmr"
+  run-python-tests:
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"] # The python versions to run tests on
+        os: ["ubuntu-latest", "macos-latest"] # The os's to run tests on
+
+    runs-on: ${{ matrix.os }}
+
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: 'pip'
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+          pip install -r tests/requirements.txt
+
+      - name: Run tests
+        run: |
+          pytest -n $pytest_cpus --cov-report=xml --cov=$pytest_cov_dir
+
+      - name: Report coverage
+        run: |
+          coverage report -m
+
+      - name: "Upload coverage to Codecov"
+        uses: codecov/codecov-action@v2
+        with:
+          fail_ci_if_error: false

tests/requirements.txt

@@ -0,0 +1,4 @@
+pytest
+pytest-xdist
+coverage
+pytest-cov

tests/test_losses.py

@@ -1,4 +1,12 @@
 """ Test loss functions"""
+
+import os
+import sys
+
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+)
+
 from dgmr.losses import SSIMLoss, MS_SSIMLoss, SSIMLossDynamic, tv_loss
 import torch
 

tests/test_model.py

@@ -1,3 +1,10 @@
+import os
+import sys
+
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+)
+
 import torch
 import torch.nn.functional as F
 from dgmr import (