From 722eb6073c327759c09d3be1e375908be7c6735b Mon Sep 17 00:00:00 2001
From: Jan-Bendikt Jagusch <jan.jagusch@quantco.com>
Date: Thu, 5 Dec 2024 13:33:47 +0100
Subject: [PATCH] Tests against both numpy versions in CI.

---
 .github/workflows/integration.yml | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
index f28c22a8..b9d8cbe8 100644
--- a/.github/workflows/integration.yml
+++ b/.github/workflows/integration.yml
@@ -7,6 +7,9 @@ on:
 
 jobs:
   run-e2e-tests:
+    strategy:
+      matrix:
+        numpy-version: [ "1.*", "2.*" ]
     runs-on: ubuntu-latest
     environment: azure-prod
     env:
@@ -53,7 +56,15 @@ jobs:
       #----------------------------------------------
       #              run test suite
       #----------------------------------------------
+      - name: Install NumPy version
+        run: |
+          source .venv/bin/activate
+          pip install "numpy==${{ matrix.numpy-version }}"
       - name: Run e2e tests
-        run: poetry run python -m pytest tests/e2e
+        run: |
+          source .venv/bin/activate
+          python -m pytest tests/e2e
       - name: Run SQL Alchemy tests
-        run: poetry run python -m pytest src/databricks/sqlalchemy/test_local
+        run: |
+          source .venv/bin/activate  
+          python -m pytest src/databricks/sqlalchemy/test_local