diff --git a/conftest.py b/conftest.py index 4dd5b00..3d6b45d 100644 --- a/conftest.py +++ b/conftest.py @@ -1,7 +1,7 @@ """Need to repeat the astropy header config here for tox.""" try: - from pytest_astropy_header.display import PYTEST_HEADER_MODULES, TESTED_VERSIONS # noqa: E501 + from pytest_astropy_header.display import PYTEST_HEADER_MODULES, TESTED_VERSIONS ASTROPY_HEADER = True except ImportError: ASTROPY_HEADER = False @@ -14,10 +14,11 @@ def pytest_configure(config): config.option.astropy_header = True # Customize the following lines to add/remove entries from the list of - # packages for which version numbers are displayed when running the tests. # noqa: E501 + # packages for which version numbers are displayed when running the tests. PYTEST_HEADER_MODULES.pop('Pandas', None) - PYTEST_HEADER_MODULES['scikit-image'] = 'skimage' + PYTEST_HEADER_MODULES.pop('h5py', None) PYTEST_HEADER_MODULES['astropy'] = 'astropy' + PYTEST_HEADER_MODULES['specutils'] = 'specutils' PYTEST_HEADER_MODULES['photutils'] = 'photutils' PYTEST_HEADER_MODULES['synphot'] = 'synphot' diff --git a/specreduce/conftest.py b/specreduce/conftest.py index e69a729..4bbe96f 100644 --- a/specreduce/conftest.py +++ b/specreduce/conftest.py @@ -12,7 +12,7 @@ from specutils import Spectrum1D, SpectralAxis try: - from pytest_astropy_header.display import PYTEST_HEADER_MODULES, TESTED_VERSIONS # noqa: E501 + from pytest_astropy_header.display import PYTEST_HEADER_MODULES, TESTED_VERSIONS ASTROPY_HEADER = True except ImportError: ASTROPY_HEADER = False @@ -122,10 +122,11 @@ def pytest_configure(config): config.option.astropy_header = True # Customize the following lines to add/remove entries from the list of - # packages for which version numbers are displayed when running the tests. # noqa: E501 + # packages for which version numbers are displayed when running the tests. PYTEST_HEADER_MODULES.pop('Pandas', None) - PYTEST_HEADER_MODULES['scikit-image'] = 'skimage' + PYTEST_HEADER_MODULES.pop('h5py', None) PYTEST_HEADER_MODULES['astropy'] = 'astropy' + PYTEST_HEADER_MODULES['specutils'] = 'specutils' PYTEST_HEADER_MODULES['photutils'] = 'photutils' PYTEST_HEADER_MODULES['synphot'] = 'synphot'