ci: add tests for the queries of TPC-H

.github/workflows/check_tpch_queries.yml

@@ -0,0 +1,28 @@
+name: Tests for TPCH Queries
+
+on:
+  pull_request:
+    types: [labeled]
+
+jobs:
+  validate-queries:
+    if: ${{ github.event.label.name == 'full-test' }}
+    strategy:
+      matrix:
+        python-version: ["3.12"]
+        os: [ubuntu-latest]
+
+    runs-on: ${{ matrix.os }}
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install uv 
+        run: curl -LsSf https://astral.sh/uv/install.sh | sh
+      - name: install-reqs
+        run: uv pip install --upgrade -r tpch/requirements-tpch.txt --system
+      - name: generate-data
+        run: cd tpch && python generate_data.py
+      - name: tpch-tests 
+        run: python -m unittest discover -s 'tpch/tests'

pyproject.toml

@@ -76,6 +76,7 @@ lint.ignore = [
 
 [tool.ruff.lint.per-file-ignores]
 "tests/*" = ["S101"]
+"tpch/tests/*" = ["S101"]
 "utils/*" = ["S311", "PTH123"]
 "tpch/execute/*" = ["T201"]
 

tpch/requirements-tpch.txt

@@ -0,0 +1,7 @@
+duckdb
+pyarrow
+polars
+pandas
+dask[dataframe]
+tqdm
+narwhals

tpch/tests/test_queries.py

@@ -0,0 +1,29 @@
+import os
+import subprocess
+import sys
+import unittest
+from pathlib import Path
+
+
+class TestQueries(unittest.TestCase):
+    def test_execute_scripts(self) -> None:
+        root = Path(__file__).resolve().parent.parent
+        # directory containing all the queries
+        execute_dir = root / "execute"
+
+        env = os.environ.copy()
+        env["PYTHONPATH"] = str(root)
+
+        for script_path in execute_dir.glob("q[1-9]*.py"):
+            result = subprocess.run(  # noqa: S603
+                [sys.executable, str(script_path)],
+                capture_output=True,
+                text=True,
+                env=env,
+                cwd=root,
+                check=False,
+                shell=False,
+            )
+            assert (
+                result.returncode == 0
+            ), f"Script {script_path} failed with error: {result.stderr}"