# Licensed under a 3-clause BSD style license - see LICENSE.rst
import os
from distutils.version import LooseVersion
# this contains imports plugins that configure py.test for astropy tests.
# by importing them here in conftest.py they are discoverable by py.test
# no matter how it is invoked within the source tree.

from astropy.version import version as astropy_version

if LooseVersion(astropy_version) < LooseVersion('2.0.3'):
    # Astropy is not compatible with the standalone plugins prior this while
    # astroquery requires them, so we need this workaround. This will mess
    # up the test header, but everything else will work.
    from astropy.tests.pytest_plugins import (PYTEST_HEADER_MODULES,
                                              TESTED_VERSIONS)
elif astropy_version < '3.0':
    # With older versions of Astropy, we actually need to import the pytest
    # plugins themselves in order to make them discoverable by pytest.
    from astropy.tests.pytest_plugins import *
else:
    # As of Astropy 3.0, the pytest plugins provided by Astropy are
    # automatically made available when Astropy is installed. This means it's
    # not necessary to import them here, but we still need to import global
    # variables that are used for configuration.
    from astropy.tests.plugins.display import (PYTEST_HEADER_MODULES,
                                               TESTED_VERSIONS)
    try:
        # The astropy test runner sets up the header automatically,
        # we should import it only when runnig pytest directly.
        _ASTROPY_TEST_
    except NameError:
        from astropy.tests.plugins.display import pytest_report_header

from astropy.tests.helper import enable_deprecations_as_exceptions

# Add astropy to test header information and remove unused packages.
# Pytest header customisation was introduced in astropy 1.0.

try:
    PYTEST_HEADER_MODULES['Astropy'] = 'astropy'
    PYTEST_HEADER_MODULES['APLpy'] = 'aplpy'
    PYTEST_HEADER_MODULES['pyregion'] = 'pyregion'
    PYTEST_HEADER_MODULES['pyVO'] = 'pyvo'
    # keyring doesn't provide __version__ any more
    # PYTEST_HEADER_MODULES['keyring'] = 'keyring'
    del PYTEST_HEADER_MODULES['h5py']
    del PYTEST_HEADER_MODULES['Scipy']
    del PYTEST_HEADER_MODULES['Pandas']
except (NameError, KeyError):
    pass

# ignoring pyvo can be removed once we require >0.9.3
enable_deprecations_as_exceptions(include_astropy_deprecations=False,
                                  warnings_to_ignore_entire_module=['pyregion'],
                                  modules_to_ignore_on_import=['pyvo'])

# add '_testrun' to the version name so that the user-agent indicates that
# it's being run in a test
from . import version
version.version += '_testrun'


# This is to figure out the affiliated package version, rather than
# using Astropy's
from .version import version, astropy_helpers_version


packagename = os.path.basename(os.path.dirname(__file__))
TESTED_VERSIONS[packagename] = version
TESTED_VERSIONS['astropy_helpers'] = astropy_helpers_version

@@ -71,51 +71,8 @@
  os%0A
-from distutils.version import LooseVersion%0A
 # th
@@ -278,363 +278,37 @@
 rom 
-astropy.version import version as astropy_version%0A%0Aif LooseVersion(astropy_version) %3C LooseVersion('2.0.3'):%0A    # Astropy is not compatible with the standalone plugins prior this while%0A    # astroquery requires them, so we need this workaround. This will mess%0A    # up the test header, but everything else will work.%0A    from astropy.tests.pytest_plugins
+pytest_astropy_header.display
  imp
@@ -370,35 +370,32 @@
                 
-   
 TESTED_VERSIONS)
@@ -399,967 +399,148 @@
 NS)%0A
-elif astropy_version %3C '3.0':%0A    # With older versions of Astropy, we actually need to import the pytest%0A    # plugins themselves in order to make them discoverable by pytest.%0A    from astropy.tests.pytest_plugins import *%0Aelse:%0A    # As of Astropy 3.0, the pytest plugins provided by Astropy are%0A    # automatically made available when Astropy is installed. This means it's%0A    # not necessary to import them here, but we still need to import global%0A    # variables that are used for configuration.%0A    from astropy.tests.plugins.display import (PYTEST_HEADER_MODULES,%0A                                               TESTED_VERSIONS)%0A    try:%0A        # The astropy test runner sets up the header automatically,%0A        # we should import it only when runnig pytest directly.%0A        _ASTROPY_TEST_%0A    except NameError:%0A        from astropy.tests.plugins.display import pytest_report_header%0A%0Afrom astropy.tests.helper import enable_deprecations_as_exceptions
+%0Afrom astropy.tests.helper import enable_deprecations_as_exceptions%0A%0A%0Adef pytest_configure(config):%0A    config.option.astropy_header = True%0A
 %0A%0A# 

Fix pytest headers