U d{jR@sddlZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZddlmZmZdd Zd d ZGd d d eZGdddeeZdS)N) Cosmology FlatLambdaCDMPlanck18)units) from_yamlto_yamlyaml_constructoryaml_representer) AstropyDumperdumpload)ToFromDirectTestBaseToFromTestMixinBasecCs<td}t|sttt}t|ts*t|ds8tdS)z9Test :func:`~astropy.cosmology.io.yaml.yaml_representer`.z!!astropy.cosmology.flrw.LambdaCDMz%!astropy.cosmology.flrw.FlatLambdaCDMN)r callableAssertionErrorr r isinstancestr startswith)Z representerymlrc/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/cosmology/io/tests/test_yaml.pytest_yaml_representers  rc Cshtt}t|sttt}ttt |}W5QRXt |tsHt|tksTt|j tj ksdtdS)z9Test :func:`~astropy.cosmology.io.yaml.yaml_constructor`.N) rrrrr ruZadd_enabled_unitscur rmeta) constructorrcosmorrrtest_yaml_constructor&s   rc@sZeZdZdZejddZddZddZdd Z ej d d d d gddZ ddZ 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. cCs |tjkrtd|ddS)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 Zyaml_representerspytestZxfail)selfZ cosmo_clsrrr!xfail_if_not_registered_with_yamlDs  z5ToFromYAMLTestMixin.xfail_if_not_registered_with_yamlcCs(|d}t|tst|ds$tdS)zTest cosmology -> YAML.yamlz!astropy.cosmology.N)rrrr)r!r to_formatr"rrrr test_to_yamlRsz ToFromYAMLTestMixin.test_to_yamlcCs`|d}||dd}|j|jks$t|j|jks4t||dd}||ksLt|j|jks\tdS)z$Test cosmology -> YAML -> cosmology.r#formatN)namerr)r!rr$ from_formatr"rgotrrrtest_from_yaml_defaultYs   z*ToFromYAMLTestMixin.test_from_yaml_defaultc Cs@|ddkst|d}tttf||W5QRXdS)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)Zcan_autodentifyrr ZraisesFileNotFoundErrorOSError)r!rr$r)r"rrrrtest_from_yaml_autoidentifylsz/ToFromYAMLTestMixin.test_from_yaml_autoidentifyr'TFNcCs4|d}t|trt|j||d}|dks0tdS)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#r&FN)rrr is_equivalent)r!rr$r'r"objZis_equivrrrtest_is_equivalent_to_yamls z.ToFromYAMLTestMixin.test_is_equivalent_to_yamlcCs|j|ddddkstdS)zTest :meth:`astropy.cosmology.Cosmology.is_equivalent`. Same as ``test_is_equivalent_to_yaml`` but with ``format="yaml"``. r#r&TN)r/r)r!rr$r"rrr)test_is_equivalent_to_yaml_specify_formatsz=ToFromYAMLTestMixin.test_is_equivalent_to_yaml_specify_format)__name__ __module__ __qualname____doc__r fixturer"r%r+r.markZ parametrizer1r2rrrrr:s   rc@s6eZdZdZddZejdddddZd d Zd S) 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. cCsttd|_dS)zBSet up fixtures to use ``to/from_yaml``, not the I/O abstractions.)tofromN)rrZ functionsr!rrr setup_classszTestToFromYAML.setup_classclassT)scopeZautouseccs dVdS)z Setup and teardown for tests. This overrides from super because `ToFromDirectTestBase` adds a custom Cosmology ``CosmologyWithKwargs`` that is not registered with YAML. Nrr<rrrsetupszTestToFromYAML.setupcCsdS)z If directly calling the function there's no auto-identification. So this overrides the test from `ToFromYAMLTestMixin` Nr)r!rr$r)rrrr.sz*TestToFromYAML.test_from_yaml_autoidentifyN) r3r4r5r6r=r r7r@r.rrrrr9s   r9)r Z astropy.unitsrrZastropy.cosmologyrrrrZastropy.cosmology.io.yamlrrrr Zastropy.io.misc.yamlr r r baserrrrrr9rrrrs   e