HR-eRddlZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZddlmZmZdZd ZGd d eZGd d eeZdS)N) Cosmology FlatLambdaCDMPlanck18)units) from_yamlto_yamlyaml_constructoryaml_representer) AstropyDumperdumpload)ToFromDirectTestBaseToFromTestMixinBasectd}t|sJtt}t |t sJ|dsJdS)z9Test :func:`~astropy.cosmology.io.yaml.yaml_representer`.z!!astropy.cosmology.flrw.LambdaCDMz%!astropy.cosmology.flrw.FlatLambdaCDMN)r callabler r isinstancestr startswith) representerymls Dlib/python3.11/site-packages/astropy/cosmology/io/tests/test_yaml.pytest_yaml_representerrsh##FGGK K     x..C c3   >>A B BBBBBBcntt}t|sJtt}t jt5t|}dddn #1swxYwYt|tsJ|tksJ|j tj ksJdS)z9Test :func:`~astropy.cosmology.io.yaml.yaml_constructor`.N) r rrr ruadd_enabled_unitscur rmeta) constructorrcosmos rtest_yaml_constructorr"&s#=11K K     x..C R S  e] + ++++ H     : & & & & & &sA//A36A3ceZdZdZejdZdZdZdZ ej dgddZ d Z d S) ToFromYAMLTestMixina Tests for a Cosmology[To/From]Format with ``format="yaml"``. 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 ``TestCosmologyToFromFormat`` or ``TestCosmology`` for examples. cV|tjvrtjd|ddSdS)z YAML I/O only works on registered classes. So the thing to check is if this class is registered. If not, :func:`pytest.xfail` this test. Some of the tests define custom cosmologies. They are not registered. zCosmologies of type z are not registered with YAML.N)r yaml_representerspytestxfail)self cosmo_clss r!xfail_if_not_registered_with_yamlz5ToFromYAMLTestMixin.xfail_if_not_registered_with_yamlDsF M; ; ; LPyPPP      < ;rcx|d}t|tsJ|dsJdS)zTest cosmology -> YAML.yamlz!astropy.cosmology.N)rrr)r)r! to_formatr+rs r test_to_yamlz ToFromYAMLTestMixin.test_to_yamlRsFi#s#####~~34444444rc|d}||d}|j|jksJ|j|jksJ||d}||ksJ|j|jksJdS)z$Test cosmology -> YAML -> cosmology.r-formatN)namer)r)r!r. from_formatr+rgots rtest_from_yaml_defaultz*ToFromYAMLTestMixin.test_from_yaml_defaultYsik#f---x5:%%%%x5:%%%%k#f---e||||x5:%%%%%%rc|ddusJ|d}tjttf5||ddddS#1swxYwYdS)zAs a non-path string, it does NOT auto-identifies 'format'. TODO! this says there should be different types of I/O registries. not just hacking object conversion on top of file I/O. r-FN)can_autodentifyr'raisesFileNotFoundErrorOSError)r)r!r.r4r+rs rtest_from_yaml_autoidentifyz/ToFromYAMLTestMixin.test_from_yaml_autoidentifyls##F++u4444i ]-w7 8 8   K                     s AA"%A"r2)TFNc|d}t|trJ|||}|dusJdS)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 YAML string. YAML can't be identified without "format" specified. r-r1FN)rr is_equivalent)r)r!r.r2r+objis_equivs rtest_is_equivalent_to_yamlz.ToFromYAMLTestMixin.test_is_equivalent_to_yamlsXic9-----&&s6&::5      rcN||dddusJdS)zTest :meth:`astropy.cosmology.Cosmology.is_equivalent`. Same as ``test_is_equivalent_to_yaml`` but with ``format="yaml"``. r-r1TN)r>)r)r!r.r+s r)test_is_equivalent_to_yaml_specify_formatz=ToFromYAMLTestMixin.test_is_equivalent_to_yaml_specify_formats8""99V#4#4V"DDLLLLLLrN)__name__ __module__ __qualname____doc__r'fixturer+r/r6r<mark parametrizerArCrrr$r$:s ^  ^ 555&&&&2 [X':':':;; ! !<; !MMMMMrr$cTeZdZdZdZejdddZdZdS) TestToFromYAMLa' Directly test ``to/from_yaml``. These are not public API and are discouraged from use, in favor of ``Cosmology.to/from_format(..., format="yaml")``, but should be tested regardless b/c 3rd party packages might use these in their Cosmology I/O. Also, it's cheap to test. c.ttd|_dS)zBSet up fixtures to use ``to/from_yaml``, not the I/O abstractions.)tofromN)rr functionsr)s r setup_classzTestToFromYAML.setup_classs ';;rclassT)scopeautousec#KdVdS)z Setup and teardown for tests. This overrides from super because `ToFromDirectTestBase` adds a custom Cosmology ``CosmologyWithKwargs`` that is not registered with YAML. NrKrRs rsetupzTestToFromYAML.setups rcdS)z If directly calling the function there's no auto-identification. So this overrides the test from `ToFromYAMLTestMixin` NrK)r)r!r.r4s rr<z*TestToFromYAML.test_from_yaml_autoidentifysrN) rDrErFrGrSr'rHrXr<rKrrrMrMsi<<<V^'400010     rrM)r' astropy.unitsrrastropy.cosmologyrrrrastropy.cosmology.io.yamlrrr r astropy.io.misc.yamlr r r baserrrr"r$rMrKrrr_s]  @@@@@@@@@@)))))) ;:::::::::;;;;;;;; C C C ' ' '(bMbMbMbMbM-bMbMbMJ     )+>     r