U e{j@sddlZddlmZddlmZmZmZddlmZddl m Z ddl m Z dde DZ Gdd d ZGd d d eZGd d d eZGdddeZGdddeeZGdddeeZdS)N) Cosmology Parameter realizations)units)_COSMOLOGY_CLASSES) availablecCsg|]}tt|qS)getattrr).0namerr^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/cosmology/io/tests/base.py sr c@seZdZdZdS) IOTestBaseaBase class for Cosmology I/O tests. This class will not be directly called by :mod:`pytest` since its name does not begin with ``Test``. To activate the contained tests this class must be inherited in a subclass. Subclasses must define a :func:`pytest.fixture` ``cosmo`` that returns/yields an instance of a |Cosmology|. See ``TestCosmology`` for an example. N)__name__ __module__ __qualname____doc__rrrr rsrc@s@eZdZdZejddddZejddddZdd Zd S) ToFromTestMixinBaseaTests for a Cosmology[To/From]Format with some ``format``. This class will not be directly called by :mod:`pytest` since its name does not begin with ``Test``. To activate the contained tests this class must be inherited in a subclass. Subclasses must define a :func:`pytest.fixture` ``cosmo`` that returns/yields an instance of a |Cosmology|. See ``TestCosmology`` for an example. classscopecCstjS)z7Convert to Cosmology using ``Cosmology.from_format()``.)r from_formatselfrrr r(szToFromTestMixinBase.from_formatcCs|jS)z2Convert Cosmology instance using ``.to_format()``.) to_formatrcosmorrr r-szToFromTestMixinBase.to_formatcCs|tjjjkS)z)Check whether a format can auto-identify.)rrregistryZ _identifiers)rformatrrr can_autodentify2sz#ToFromTestMixinBase.can_autodentifyN) rrrrpytestfixturerrrrrrr rs     rc@sFeZdZdZejddddZejddddZejdd Zd S) ReadWriteTestMixinBaseaTests for a Cosmology[Read/Write]. This class will not be directly called by :mod:`pytest` since its name does not begin with ``Test``. To activate the contained tests this class must be inherited in a subclass. Subclasses must define a :func:`pytest.fixture` ``cosmo`` that returns/yields an instance of a |Cosmology|. See ``TestCosmology`` for an example. rrcCstjS)z3Read Cosmology instance using ``Cosmology.read()``.)rreadrrrr r#AszReadWriteTestMixinBase.readcCs|jS)z#Write Cosmology using ``.write()``.)writerrrr r$FszReadWriteTestMixinBase.writec cs tt dVW5QRXdS)z8Add :mod:`astropy.cosmology.units` to the enabled units.N)uZadd_enabled_unitscurrrr add_cuKs zReadWriteTestMixinBase.add_cuN) rrrrr r!r#r$r'rrrr r"7s    r"c@sPeZdZdZejdddddZejdeddd Zejdd d d Z d S)IODirectTestBaseaDirectly test Cosmology I/O functions. These functions are not public API and are discouraged from public use, in favor of the I/O methods on |Cosmology|. They are tested b/c they are used internally and because some tests for the methods on |Cosmology| don't need to be run in the |Cosmology| class's large test matrix. This class will not be directly called by :mod:`pytest` since its name does not begin with ``Test``. To activate the contained tests this class must be inherited in a subclass. rT)rZautouseccs(Gdddt}dVt|jddS)zSetup and teardown for tests.cs*eZdZeejdZdfdd ZZS)z3IODirectTestBase.setup..CosmologyWithKwargs)unitrcosmology with kwargsNcs tj||d|tj>|_dS)N)r meta)super__init__r%KZ_Tcmb0)rTcmb0r r+kwargs __class__rr r-jsz.CosmologyWithKwargs.__init__)rr*N) rrrrr%r.r/r- __classcell__rrr1r CosmologyWithKwargsgs  r4N)rrpopr)rr4rrr setupcs zIODirectTestBase.setup)rparamscCs"t|jtrt|jddS|jS)zCosmology instance.)r/) isinstanceparamstrr)rrequestrrr rvs zIODirectTestBase.cosmorcCs|jS)zCosmology classes.r1rrrr cosmo_cls}szIODirectTestBase.cosmo_clsN) rrrrr r!r6cosmo_instancesrr=rrrr r(Vs     r(c@s8eZdZdZejddddZejddddZdS) ToFromDirectTestBaseaDirectly test ``to/from_``. These functions are not public API and are discouraged from public use, in favor of ``Cosmology.to/from_format(..., format="")``. They are tested because they are used internally and because some tests for the methods on |Cosmology| don't need to be run in the |Cosmology| class's large test matrix. This class will not be directly called by :mod:`pytest` since its name does not begin with ``Test``. To activate the contained tests this class must be inherited in a subclass. Subclasses should have an attribute ``functions`` which is a dictionary containing two items: ``"to"=`` and ``"from"=``. rrcsfdd}|S)z-Convert to Cosmology using function ``from``.cs|ddjd||S)Nrfromr5 functionsargsr0rrr use_from_formats z9ToFromDirectTestBase.from_format..use_from_formatr)rrErrr rs z ToFromDirectTestBase.from_formatcsfdd}|S)z2Convert Cosmology to format using function ``to``.csjdf||S)NtorBrCrrrr use_to_formatsz5ToFromDirectTestBase.to_format..use_to_formatr)rrrIrrHr rszToFromDirectTestBase.to_formatN)rrrrr r!rrrrrr r?s   r?c@s8eZdZdZejddddZejddddZdS) ReadWriteDirectTestBaseaDirectly test ``read/write_``. These functions are not public API and are discouraged from public use, in favor of ``Cosmology.read/write(..., format="")``. They are tested because they are used internally and because some tests for the methods on |Cosmology| don't need to be run in the |Cosmology| class's large test matrix. This class will not be directly called by :mod:`pytest` since its name does not begin with ``Test``. To activate the contained tests this class must be inherited in a subclass. Subclasses should have an attribute ``functions`` which is a dictionary containing two items: ``"read"=`` and ``"write"=``. rrcsfdd}|S)z1Read Cosmology from file using function ``read``.cs|ddjd||S)Nrr#rArCrrr use_reads z.ReadWriteDirectTestBase.read..use_readr)rrKrrr r#s zReadWriteDirectTestBase.readcsfdd}|S)z1Write Cosmology to file using function ``write``.csjdf||S)Nr$rGrCrHrr use_writesz0ReadWriteDirectTestBase.write..use_writer)rrrLrrHr r$szReadWriteDirectTestBase.writeN)rrrrr r!r#r$rrrr rJs   rJ)r Z astropy.unitsrr%Zastropy.cosmologyrrrr&Zastropy.cosmology.corerZastropy.cosmology.realizationsrr>rrr"r(r?rJrrrr s     -&