fix coverage

Author: huppd

.coveragerc

@@ -1,8 +1,7 @@
 [run]
 source =
-	engine/
-    visualize/
-    probtest.py
-	util/
+  engine
+  util
+  visualize
 omit =
-	tests/*
+  tests/*

.github/workflows/pytest.yml

@@ -7,27 +7,35 @@ on:
   pull_request:
     branches:
     - main
+
 jobs:
   probtest-pytest:
     runs-on: ubuntu-latest
     defaults:
       run:
         shell: bash -l {0}
+
     steps:
     - uses: actions/checkout@v2
+
     - name: Set up Python
       uses: actions/setup-python@v2
       with:
         python-version: 3.10.8
-    - name: Install dependencies with pip
+    - name: Install Poetry
+      run: |
+        curl -sSL https://install.python-poetry.org | python3 -
+    - name: Configure Poetry to use the right Python version
+      run: |
+        poetry env use python3.10
+    - name: Install dependencies with Poetry
       run: |
-        python -m pip install --upgrade pip
-        pip install -r requirements.txt
-    - name: Run setup_env.py script
+        poetry install
+    - name: Run setup_env for eccodes definition
       run: |
-        python setup_env.py
-    - name: Run Pytest
+        poetry run python setup_env
+    - name: Run Pytest with Poetry
       env:
         TZ: Europe/Zurich
       run: |
-        pytest -v -s --cov --cov-report=term tests/
+        poetry run pytest -v -s --cov --cov-report=term tests/

setup_env.py

@@ -2,60 +2,73 @@
 """ Setup .env with local eccodes definition """
 
 import os
+import shutil
 import subprocess
 
-base_path = os.getcwd()
-
-eccodes_definition_path = os.path.join(base_path, "resource", "eccodes")
-cosmo_definition_path = os.path.join(base_path, "resource", "eccodes-cosmo-resources")
-
-# Clone repositories
-subprocess.run(
-    [
-        "git",
-        "clone",
-        "--depth",
-        "1",
-        "-b",
-        "2.39.0",
-        "https://github.com/ecmwf/eccodes.git",
-        eccodes_definition_path,
-    ],
-    check=True,
-)
-subprocess.run(
-    [
-        "git",
-        "clone",
-        "--depth",
-        "1",
-        "-b",
-        "v2.36.0.3",
-        "https://github.com/COSMO-ORG/eccodes-cosmo-resources.git",
-        cosmo_definition_path,
-    ],
-    check=True,
-)
-
-# Remove deprecated directory
-deprecated_path = os.path.join(eccodes_definition_path, "deprecated")
-if os.path.exists(deprecated_path):
-    for root, dirs, files in os.walk(deprecated_path, topdown=False):
-        for name in files:
-            os.remove(os.path.join(root, name))
-        for name in dirs:
-            os.rmdir(os.path.join(root, name))
-    os.rmdir(deprecated_path)
-
-# Remove existing .env file
-ENV_FILE = ".env"
-if os.path.exists(ENV_FILE):
-    os.remove(ENV_FILE)
-
-# Write to .env file
-ENV_CONTENT = (
-    f"ECCODES_DEFINITION_PATH={eccodes_definition_path}/definitions:"
-    f"{cosmo_definition_path}/definitions\n"
-)
-with open(ENV_FILE, "w", encoding="utf-8") as file:
-    file.write(ENV_CONTENT)
+
+def main():
+    base_path = os.getcwd()
+
+    resource_path = os.path.join(base_path, "resource")
+
+    eccodes_definition_path = os.path.join(resource_path, "eccodes")
+    cosmo_definition_path = os.path.join(resource_path, "eccodes-cosmo-resources")
+
+    # remove existing resource
+    if os.path.exists(resource_path):
+        shutil.rmtree(resource_path)
+
+    # Clone repositories
+    subprocess.run(
+        [
+            "git",
+            "clone",
+            "--depth",
+            "1",
+            "-b",
+            "2.39.0",
+            "https://github.com/ecmwf/eccodes.git",
+            eccodes_definition_path,
+        ],
+        check=True,
+    )
+    subprocess.run(
+        [
+            "git",
+            "clone",
+            "--depth",
+            "1",
+            "-b",
+            "v2.36.0.3",
+            "https://github.com/COSMO-ORG/eccodes-cosmo-resources.git",
+            cosmo_definition_path,
+        ],
+        check=True,
+    )
+
+    # Remove deprecated directory
+    deprecated_path = os.path.join(eccodes_definition_path, "deprecated")
+    if os.path.exists(deprecated_path):
+        for root, dirs, files in os.walk(deprecated_path, topdown=False):
+            for name in files:
+                os.remove(os.path.join(root, name))
+            for name in dirs:
+                os.rmdir(os.path.join(root, name))
+        os.rmdir(deprecated_path)
+
+    # Remove existing .env file
+    env_file = ".env"
+    if os.path.exists(env_file):
+        os.remove(env_file)
+
+    # Write to .env file
+    env_content = (
+        f"ECCODES_DEFINITION_PATH={eccodes_definition_path}/definitions:"
+        f"{cosmo_definition_path}/definitions\n"
+    )
+    with open(env_file, "w", encoding="utf-8") as file:
+        file.write(env_content)
+
+
+if __name__ == "__main__":
+    main()