File: conftest.py

package info (click to toggle)
astroml 0.4.post1-6
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 892 kB
  • sloc: python: 5,492; makefile: 3
file content (27 lines) | stat: -rw-r--r-- 844 bytes parent folder | download
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