HR-e)& ddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZmZddlmZejjZdd d efd hZd efd efdefdefdefgZGdde j e j!ej"Z#Gdde#Z$Gdde j%ej&ej'ej(ej)ej*Z+Gdde+Z,dS)N) cosmology) Cosmologyw0wzCDM)readwrite_registry) test_cosmology test_ecsv test_html test_json test_mapping test_modeltest_row test_table test_yaml)QTableRow)HAS_BS4)z ascii.ecsvTTz ascii.htmlF)jsonTTmappingyamlzastropy.cosmologyz astropy.rowz astropy.tableceZdZdZejdedZejdedZ dS)ReadWriteTestMixina Tests for a CosmologyRead/Write on a |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 ``TestReadWriteCosmology`` or ``TestCosmology`` for examples. zformat, metaio, has_depscz|stjdt||jd|z }|||t j|sJtjt5|||ddddn #1swxYwYt j|sJ|||dtj ||}||ksJ| t|j t|j kz sJdS)a Test writing from an instance and reading from the base class. This requires full information. The round-tripped metadata can be in a different order, so the OrderedDict must be converted to a dict before testing equality. missing a dependency.formatF)r overwriteNT)pytestskipstrnamewriteospathexistsraisesIOErrorrreaddictmeta)selfcosmotmp_pathrmetaiohas_depsfnamegots Dlib/python3.11/site-packages/astropy/cosmology/tests/test_connect.pytest_readwrite_complete_infoz/ReadWriteTestMixin.test_readwrite_complete_infoDsu 0 K. / / /H%*77v77788 E& )))w~~e$$$$$ ]7 # # ? ? KKfK > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?w~~e$$$$$ E&D 999nU6222e|||| tCH~~ej1A1AABBBBBBsB))B-0B-c||stjdt||jd|z }||||||}||ksJ| t |jt |jkz sJtj|||}||ksJ| t |jt |jkz sJtj|||j }||ksJ| t |jt |jkz sJdS)zy Test writing from an instance and reading from that class, when there's full information saved. rrrrrN) rrr r!r"r(r)r*r __qualname__) r+ cosmo_clsr,r-rr.r/r0r1s r2*test_readwrite_from_subclass_complete_infoz=ReadWriteTestMixin.test_readwrite_from_subclass_complete_info`sJ 0 K. / / /H%*77v77788 E& )))nnU6n22e|||| tCH~~ej1A1AABBBBnU6YGGGe|||| tCH~~ej1A1AABBBBnU6Y=STTTe|||| tCH~~ej1A1AABBBBBBN) __name__ __module__r6__doc__rmark parametrizereadwrite_formatsr3r8r9r2rr6s  [79JKKCCLKC6 [79JKKCCLKCCCr9rc eZdZdZejdedZejddZej de dZ ej de d Z d S) TestCosmologyReadWritez%Test the classes CosmologyRead/Write.classscopeparamsc@ttj|jSNgetattrr realizationsparamr+requests r2r,zTestCosmologyReadWrite.cosmoy-w}===r9rEc|jSrH __class__r+r,s r2r7z TestCosmologyReadWrite.cosmo_cls r9zformat, _, has_depsc|stjdtj|t}t j|}d|jvsJd|jvsJdS)Nrrzoverwrite : bool) rrr get_writerrinspect signature parametersr<)r+r_r/writersigs r20test_write_methods_have_explicit_kwarg_overwritezGTestCosmologyReadWrite.test_write_methods_have_explicit_kwarg_overwritesk 0 K. / / /#.vyAA''cn,,,,"V^333333r9cD|stjd||jd|z }|||tjt d5t j||dddn #1swxYwYtjt d5tj||t dddn #1swxYwYtjtd5t j||d ddddS#1swxYwYdS) z2Test when the reader class doesn't match the file.rrrzmissing 1 requiredmatchNr5`cosmology` must be either FlatLambdaCDM) rrr!r"r& TypeErrorrr(r ValueError)r+r,r-rr[r/r0s r2$test_readwrite_reader_class_mismatchz;TestCosmologyReadWrite.test_readwrite_reader_class_mismatchs   0 K. / / /ej336333 E& )))]9,@ A A A / / Lv . . . . / / / / / / / / / / / / / / /]9,@ A A A D D N57 C C C C D D D D D D D D D D D D D D D]:-I J J J J J Lv I I I I J J J J J J J J J J J J J J J J J Js6A;;A?A?!C  CC0DDDN)r:r;r6r<rfixturecosmo_instancesr,r7r=r>r?r^rfr@r9r2rBrB~s//V^'/:::>>;:>V^'"""#"  [24EFF 4 4GF 4 [24EFFJJGFJJJr9rBceZdZdZejdedZejdedZ dS)ToFromFormatTestMixina Tests for a Cosmology[To/From]Format on a |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. zformat, totypec||}t||sJtj||}||rtj|}||ksJ||ksJ|j|jksJdS)zRead tests happen later.rN) to_format isinstancer from_formatcan_autodentifyr*)r+r,rtotype!xfail_if_not_registered_with_yamlobjr1got2s r2test_tofromformat_complete_infoz5ToFromFormatTestMixin.test_tofromformat_complete_infos oof%%#v&&&&&#C777    ' ' (--D3;;;;e||||x5:%%%%%%r9c||}t||sJ|||}||rt j|}||ksJ||ksJ|j|jksJt j|||}||ksJ|j|jksJt j|||j}||ksJ|j|jksJdS)z Test transforming an instance and parsing from that class, when there's full information available. Partial information tests are handled in the Mixin super classes. rr5N)rlrmrnrorr*r6) r+r7r,rrprqrrr1rss r2&test_fromformat_subclass_complete_infoztofrom_formatsrtrvr@r9r2rjrjs| [-~>>&&?>&& [-~>> & &?> & & &r9rjceZdZdZejdedZejddZej de dZ d S) TestCosmologyToFromFormatz&Test Cosmology[To/From]Format classes.rCrDc@ttj|jSrHrIrMs r2r,zTestCosmologyToFromFormat.cosmorOr9rPc|jSrHrRrTs r2r7z#TestCosmologyToFromFormat.cosmo_clsrUr9 format_typec|\}}||}t||sJtjt5t j||dddn #1swxYwYtjt5tj||t dddn #1swxYwYtjtd5t j||dddddS#1swxYwYdS)Nrr5rbr`rc) rlrmrr&rdrrnrre)r+r,r|rrprrs r2test_fromformat_class_mismatchz8TestCosmologyToFromFormat.test_fromformat_class_mismatch s$oof%%#v&&&&&]9 % % 4 4  F 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4]9 % % I I  !#f H H H H I I I I I I I I I I I I I I I]:-I J J J O O  Fo N N N N O O O O O O O O O O O O O O O O O Os6A))A-0A- B66B:=B:DDDN) r:r;r6r<rrgrhr,r7r=r>rwr~r@r9r2ryrys00V^'/:::>>;:>V^'"""#"  []N;;OO<;OOOr9ry)-rXr#rastropyrastropy.cosmologyrrastropy.cosmology.connectrastropy.cosmology.io.testsrrr r r r r rr astropy.tablerr"astropy.utils.compat.optional_depsrrK availablerhr?r)r rwReadWriteECSVTestMixinReadWriteHTMLTestMixinReadWriteJSONTestMixinrrBToFromCosmologyTestMixinToFromMappingTestMixinToFromModelTestMixinToFromRowTestMixinToFromTableTestMixinToFromYAMLTestMixinrjryr@r9r2rsb 00000000888888                      &%%%%%%%666666 (2 5'" SM)$Cf ECECECECEC $ $ $ECECECP/J/J/J/J/J//J/J/JlF&F&F&F&F&+'# # ! F&F&F&ROOOOO 5OOOOOr9