add single-backend test decorators

bayesflow-org · May 28, 2024 · e4e6462 · e4e6462
1 parent b767e12
commit e4e6462
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/pytest.ini b/pytest.ini
@@ -0,0 +1,6 @@
+[pytest]
+markers =
+    jax: mark test as requiring JAX
+    numpy: mark test as requiring NumPy
+    tensorflow: mark test as requiring TensorFlow
+    torch: mark test as requiring PyTorch
diff --git a/tox.ini b/tox.ini
@@ -35,4 +35,7 @@ set_env =
         PIP_EXTRA_INDEX_URL = https://download.pytorch.org/whl/cpu
 
 commands =
-    python -m pytest tests/ -n auto -v
+    jax: python -m pytest tests/ -n auto -v -m jax
+    numpy: python -m pytest tests/ -n auto -v -m numpy
+    tensorflow: python -m pytest tests/ -n auto -v -m tensorflow
+    torch: python -m pytest tests/ -n auto -v -m torch