HR-eddlZddlmZddlmZmZmZddlmZddl m Z ddl m Z de DZ GddZGd d eZGd d eZGd deZGddeeZGddeeZdS)N) Cosmology Parameter realizations)units)_COSMOLOGY_CLASSES) availablec8g|]}tt|S)getattrr).0names ?lib/python3.11/site-packages/astropy/cosmology/io/tests/base.py r s"EEE47<..EEEceZdZdZdS) 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__r rrrrsrrceZdZdZejddZejddZdZdS)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. classscopectjS)z7Convert to Cosmology using ``Cosmology.from_format()``.)r from_formatselfs rrzToFromTestMixinBase.from_format(s $$rc|jS)z2Convert Cosmology instance using ``.to_format()``.) to_formatrcosmos rr!zToFromTestMixinBase.to_format- rc2|tjjjvS)z)Check whether a format can auto-identify.)rrregistry _identifiers)rformats rcan_autodentifyz#ToFromTestMixinBase.can_autodentify2s.7DDDrN) rrrrpytestfixturerr!r)r rrrrsV^'"""%%#"%V^'"""#"EEEEErrceZdZdZejddZejddZejdZdS)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. rrctjS)z3Read Cosmology instance using ``Cosmology.read()``.)rreadrs rr/zReadWriteTestMixinBase.readAs ~rc|jS)z#Write Cosmology using ``.write()``.)writer"s rr1zReadWriteTestMixinBase.writeFs {rc#vKtjt5dVddddS#1swxYwYdS)z8Add :mod:`astropy.cosmology.units` to the enabled units.N)uadd_enabled_unitscurs radd_cuzReadWriteTestMixinBase.add_cuKs  $ $   EEE                  s .22N) rrrrr*r+r/r1r6r rrr-r-7sV^'"""#"V^'"""#" ^^rr-ceZdZdZejdddZejdedZejdd 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)rautousec#lKGddt}dVtj|jddS)zSetup and teardown for tests.cDeZdZeejZ dfd ZxZS)3IODirectTestBase.setup..CosmologyWithKwargs)unitrcosmology with kwargsNcvt|||tjz|_dS)N)r meta)super__init__r3K_Tcmb0)rTcmb0r r@kwargs __class__s rrBz.CosmologyWithKwargs.__init__js3  d 666#qsl r)rr>N) rrrrr3rCrErB __classcell__rGs@rCosmologyWithKwargsr<gsXI13'''ECG + + + + + + + + + +rrJN)rrpopr)rrJs rsetupzIODirectTestBase.setupcsZ + + + + +) + + +  2?FFFFFr)rparamsczt|jtrt|jdS|jS)zCosmology instance.)rE) isinstanceparamstrr)rrequests rr#zIODirectTestBase.cosmovs8 gmS ) ) >%gm41=== =}rrc|jS)zCosmology classes.rIr"s r cosmo_clszIODirectTestBase.cosmo_cls}r$rN) rrrrr*r+rLcosmo_instancesr#rUr rrr8r8Vs  V^'4000GG10G$V^'/:::;: V^'"""#"rr8czeZdZdZejddZejddZdS)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"=``. rrcfd}|S)z-Convert to Cosmology using function ``from``.cV|ddjd|i|S)Nr(fromrK functionsargsrFrs ruse_from_formatz9ToFromDirectTestBase.from_format..use_from_format3 JJx & & &)4>&)4:6:: :rr )rr`s` rrz ToFromDirectTestBase.from_formats$ ; ; ; ; ;rcfd}|S)z2Convert Cosmology to format using function ``to``.c2jdg|Ri|S)Ntor]r_rFr#rs r use_to_formatz5ToFromDirectTestBase.to_format..use_to_formats*'4>$'?????? ?rr )rr#rgs`` rr!zToFromDirectTestBase.to_formats0 @ @ @ @ @ @rN)rrrrr*r+rr!r rrrXrXst"V^'"""#"V^'"""#"rrXczeZdZdZejddZejddZdS)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"=``. rrcfd}|S)z1Read Cosmology from file using function ``read``.cV|ddjd|i|S)Nr(r/r\r^s ruse_readz.ReadWriteDirectTestBase.read..use_readrarr )rrls` rr/zReadWriteDirectTestBase.reads# ; ; ; ; ;rcfd}|S)z1Write Cosmology to file using function ``write``.c2jdg|Ri|S)Nr1rerfs r use_writez0ReadWriteDirectTestBase.write..use_writes**4>'*5B4BBB6BB Brr )rr#ros`` rr1zReadWriteDirectTestBase.writes0 C C C C C CrN)rrrrr*r+r/r1r rrririst"V^'"""#"V^'"""#"rri)r* astropy.unitsrr3astropy.cosmologyrrrr5astropy.cosmology.corerastropy.cosmology.realizationsrrVrrr-r8rXrir rrrts @@@@@@@@@@))))))555555444444EE9EEE EEEEE*EEE2Z>*****z***Z#####+-@###L#####.0C#####r