1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
|
from pytest_astropy_header.display import (PYTEST_HEADER_MODULES,
TESTED_VERSIONS)
def pytest_configure(config):
config.option.astropy_header = True
try:
PYTEST_HEADER_MODULES['Cython'] = 'Cython'
PYTEST_HEADER_MODULES['Astropy'] = 'astropy'
PYTEST_HEADER_MODULES['scikit-learn'] = 'sklearn'
# pymc import here triggers pytest INTERNALERROR thus as a temporary
# measure we don't list its version number
# PYTEST_HEADER_MODULES['pymc'] = 'pymc'
PYTEST_HEADER_MODULES['astroML_addons'] = 'astroML_addons'
del PYTEST_HEADER_MODULES['h5py']
del PYTEST_HEADER_MODULES['Pandas']
except (KeyError):
pass
# This is to figure out the package version, rather than
# using Astropy's
from . import __version__ as version
TESTED_VERSIONS['astroMl'] = version
|