diff --git a/conftest.py b/conftest.py new file mode 100644 index 0000000..db5cbf4 --- /dev/null +++ b/conftest.py @@ -0,0 +1,31 @@ +# Copyright (C) 2024 Jelle van der Werff +# +# This file is part of thebeat. +# +# thebeat is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# thebeat is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with thebeat. If not, see . + +import sys + +import pytest + +NUMPY2_DOCTESTS = [ + 'thebeat.stats.ks_test' +] + + +def pytest_collection_modifyitems(items): + skipif = pytest.mark.skipif(condition=sys.version_info < (3, 9), reason="Doctest requires NumPy 2 to be installed") + for item in items: + if item.name in NUMPY2_DOCTESTS: + item.add_marker(skipif) diff --git a/tests/requirements.txt b/tests/requirements.txt index 8d4d82c..baa3a79 100644 --- a/tests/requirements.txt +++ b/tests/requirements.txt @@ -1,2 +1,5 @@ pytest pytest-mpl + +numpy>=2; python_version>=3.9 +numpy<2; python_version==3.8