
    HR-eQ                     V    d Z ddlZddlZddlmZ ddgZd	dZej        d             ZdS )
z
This module provides the tools used to internally run the cosmology test suite
from the installed astropy.  It makes use of the `pytest`_ testing framework.
    N)coreget_redshift_methodsclean_registryTc                    t                      }t          |           D ]_}	 t          | |          }n# t          $ r Y  w xY wt	          |          r,|                    d          r|r|                    |           `t          j        |           rdnd}t          |          D ]-}	 t          j
        t          | |                    }t          |j                                                  }n%# t          $ r |                    |           Y nw xY wt!          |          |k    r|                    |           t!          |          |dz   k    r1||                             d          s|                    |           |sIt!          |          |dz   k    r3||dz                                d          r|                    |           /|S )a  Get redshift methods from a cosmology.

    Parameters
    ----------
    cosmology : |Cosmology| class or instance
    include_private : bool
        Whether to include private methods, i.e. starts with an underscore.
    include_z2 : bool
        Whether to include methods that are functions of 2 (or more) redshifts,
        not the more common 1 redshift argument.

    Returns
    -------
    set[str]
        The names of the redshift methods on `cosmology`, satisfying
        `include_private` and `include_z2`.
    _   r   z   )setdirgetattrNotImplementedErrorcallable
startswithaddinspectisclasstuple	signaturelist
parameterskeys
ValueErrordiscardlen)		cosmologyinclude_private
include_z2methodsnmiz1sigparamss	            >lib/python3.11/site-packages/astropy/cosmology/tests/helper.pyr   r      s   & eeG^^  		1%%AA" 	 	 	H	 A;; 	S 1 1 	_ 	KKNNN
 y))
0!!qC7^^  	1#GIq$9$9::C
 #.--//00FF	  	 	 	OOAH	 v;;#OOA[[C!G##F3K,B,B-
 -
# OOA 	Va''sQw**3// ( OOANs   2
??)"C22DDc               #   n   K   t           j        } i t           _        t           j        V  | t           _        dS )zC`pytest.fixture` for clearing and restoring ``_COSMOLOGY_CLASSES``.N)r   _COSMOLOGY_CLASSES)ORIGINAL_COSMOLOGY_CLASSESs    r%   r   r   [   s6       "&!8 D

!!!!8D    )TT)	__doc__r   pytestastropy.cosmologyr   __all__r   fixturer    r)   r%   <module>r0      s       # " " " " "!#3
4: : : :B 9 9 9 9 9r)   