Clean up test header

astropy · Dec 5, 2023 · 8508eeb · 8508eeb
1 parent 9c9418d
commit 8508eeb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git 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
@@ -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'