pytest-quick

Author: mpharrigan

.github/workflows/nightly.yaml

@@ -43,4 +43,4 @@ jobs:
           pip install -r dev_tools/requirements/envs/pytest.env.txt
           pip install --no-deps -e .
       - run: |
-          check/pytest --durations=10 -v -n=0
+          check/pytest-full --durations=10 -v -n=0

.github/workflows/pr.yaml

@@ -41,7 +41,7 @@ jobs:
           pip install -r dev_tools/requirements/envs/pytest.env.txt
           pip install --no-deps -e .
       - run: |
-          check/pytest -m 'not notebook and not slow'
+          check/pytest-quick
 
   dev-tools:
     runs-on: ubuntu-24.04

check/pytest

@@ -13,6 +13,12 @@ thisdir="$(dirname "${BASH_SOURCE[0]}")" || exit $?
 topdir="$(git -C "${thisdir}" rev-parse --show-toplevel)" || exit $?
 cd "${topdir}" || exit $?
 
+echo "Warning: running the full test suite, which may take tens of minutes."
+echo ""
+echo "  To run the recommended test suite for development: use check/pytest-quick"
+echo "  To suppress this warning: use check/pytest-full"
+echo ""
+
 # Run in parallel by default.  Pass the `-n0` option for a single-process run.
 # (the last `-n` option wins)
 PYTEST_ARGS=( "-n=auto" "$@" )

check/pytest-full

@@ -0,0 +1,14 @@
+#!/usr/bin/env bash
+
+# Get the working directory to the repo root.
+cd "$( dirname "${BASH_SOURCE[0]}" )" || exit 1
+cd "$(git rev-parse --show-toplevel)" || exit 1
+
+# Run in parallel by default.  Pass the `-n0` option for a single-process run.
+# (the last `-n` option wins)
+PYTEST_ARGS=( "-n=auto" "$@" )
+
+pytest "${PYTEST_ARGS[@]}" qualtran/
+RESULT=$?
+
+exit "$RESULT"

check/pytest-quick

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+
+# Get the working directory to the repo root.
+cd "$( dirname "${BASH_SOURCE[0]}" )" || exit 1
+cd "$(git rev-parse --show-toplevel)" || exit 1
+
+# Run in parallel by default.  Pass the `-n0` option for a single-process run.
+# (the last `-n` option wins)
+PYTEST_ARGS=( "-n=auto" "-m" 'not slow and not notebook' "$@" )
+pytest "${PYTEST_ARGS[@]}" qualtran/
+RESULT=$?
+
+exit "$RESULT"