diff --git a/pytest.ini b/pytest.ini new file mode 100644 index 00000000..1e787e5b --- /dev/null +++ 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 index 5c9c8e4f..1d9ded2e 100644 --- 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