diff --git a/.azure/azure-ci-build-pipeline.yaml b/.azure/azure-ci-build-pipeline.yaml
index 01111290..2375a391 100644
--- a/.azure/azure-ci-build-pipeline.yaml
+++ b/.azure/azure-ci-build-pipeline.yaml
@@ -102,9 +102,7 @@ stages:
               version: '2.1.x'
 
           - script: |
-              coverage run -m --source=primaite pytest -v -o junit_family=xunit2 --junitxml=junit/test-results.xml --cov-fail-under=80
-              coverage xml -o coverage.xml -i
-              coverage html -d htmlcov -i
+              python run_test_and_coverage.py
             displayName: 'Run tests and code coverage'
 
           # Run the notebooks
diff --git a/run_test_and_coverage.py b/run_test_and_coverage.py
new file mode 100644
index 00000000..3bd9072d
--- /dev/null
+++ b/run_test_and_coverage.py
@@ -0,0 +1,22 @@
+# © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK
+import subprocess
+import sys
+from typing import Any
+
+
+def run_command(command: Any):
+    """Runs a command and returns the exit code."""
+    result = subprocess.run(command, shell=True)
+    if result.returncode != 0:
+        sys.exit(result.returncode)
+
+
+# Run pytest with coverage
+run_command(
+    "coverage run -m --source=primaite pytest -v -o junit_family=xunit2 "
+    "--junitxml=junit/test-results.xml --cov-fail-under=80"
+)
+
+# Generate coverage reports if tests passed
+run_command("coverage xml -o coverage.xml -i")
+run_command("coverage html -d htmlcov -i")