U e{j @sNddlZddlmZmZddlmZmZGdddeZGdddeeZdS) N)from_cosmology to_cosmology)IODirectTestBaseToFromTestMixinBasec@sFeZdZdZddZddZddZej dd d d d gd dZ d S)ToFromCosmologyTestMixina Tests for a Cosmology[To/From]Format with ``format="astropy.cosmology"``. 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. cCs|d}||kstdS)Test cosmology -> cosmology.astropy.cosmologyNAssertionError)selfcosmo to_formatnewcosmorh/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/cosmology/io/tests/test_cosmology.pytest_to_cosmology_defaultsz2ToFromCosmologyTestMixin.test_to_cosmology_defaultc Cs&tt|dddW5QRXdS)z%Test incorrect type in ``Cosmology``.zNOT A COSMOLOGYr formatN)pytestZraises TypeError)r r from_formatrrrtest_from_not_cosmologys z0ToFromCosmologyTestMixin.test_from_not_cosmologycCs||}||kstdS)rNr )r r rrrrrtest_from_cosmology_default"sz4ToFromCosmologyTestMixin.test_from_cosmology_defaultrTFNr cCs2|d}||kst|j||d}|dks.tdS)a`Test :meth:`astropy.cosmology.Cosmology.is_equivalent`. This test checks that Cosmology equivalency can be extended to any Python object that can be converted to a Cosmology -- in this case a Cosmology! Since it's the identity conversion, the cosmology is always equivalent to itself, regardless of ``format``. r rTN)r Z is_equivalent)r r rrobjZis_equivrrrtest_is_equivalent_to_cosmology's  z8ToFromCosmologyTestMixin.test_is_equivalent_to_cosmology) __name__ __module__ __qualname____doc__rrrrmarkZ parametrizerrrrrrs  rc@seZdZdZddZdS)TestToFromCosmologyz$Directly test ``to/from_cosmology``.cCsttd|_dS)N)tofrom)rrZ functions)r rrr setup_class:szTestToFromCosmology.setup_classN)rrrrr$rrrrr!7sr!) rZastropy.cosmology.io.cosmologyrrbaserrrr!rrrrs)