HR-eM4ddlZddlZddlmZddlZddlZddlmZ ddl m Z m Z ddl mZmZddlmZmZmZmZmZmZmZmZddlmZddlmZdd lmZej e d d Z!ej e d Gd dZ"ej e d dZ#ej e d dZ$ej e d ej%dgddZ&ej e d ej%dgddZ'ej e d ej%deeeeeeeee j(e j)e j*e j+e j,e j-e j.e j/gdZ0dS)N)StringIO)units)coreflrw)_z_at_scalar_value z_at_value)WMAP1WMAP3WMAP5WMAP7WMAP9Planck13Planck15Planck18)allclose) HAS_SCIPY)AstropyUserWarningztest requires scipy)reasonc.t}tt|jdtjzddsJtt|jdtjzddsJtt|jdtjzddsJtt|j d tj zd dsJtt|j d tj zd ddsJtt|j dtj zdddsJtt|j dtj zdddsJtjtj5tjt$d5t|j dtj zddddn #1swxYwYdddn #1swxYwYtjtj5tjt$d5t|j dtj zddddn #1swxYwYddddS#1swxYwYdS)Ng&W @ư>rtolgCr?.gʲb?g@g=*?g9{ :@g|=)ztolg& .>)zmax,?@)zminԭ T@fval is not bracketedmatch?@)rrrageuGyr lookback_timedistmodmagluminosity_distanceMpcGpcangular_diameter_distancepytestraisesrCosmologyErrorwarnsr)cosmos Hlib/python3.11/site-packages/astropy/cosmology/funcs/tests/test_funcs.pytest_z_at_value_scalarr9s E Juy!ae)44jt L L LLLL Ju2AI>> RV W W WWWW Ju}b15j99;T R R RRRR 5,cAEk::JT    5,lQU.BOOO     52D15LqIII     52D15LsKKK     t* + +PP \,4L M M M P P u6qu 3 O O O O P P P P P P P P P P P P P P PPPPPPPPPPPPPPPP t* + +PP \,4L M M M P P u6qu 3 O O O O P P P P P P P P P P P P P P PPPPPPPPPPPPPPPPPPPslG<4%G% G<%G) )G<,G) -G<<HH%J %I2& J 2I6 6J 9I6 :J  JJc,eZdZdZdZdZdZdZdS) Test_ZatValuect|_dSN)rr7selfs r8 setup_classzTest_ZatValue.setup_classHs  c tt|jjddgtjzddgdsJtt|jjdtjzdd gdd g d d gdsJtt|jjdtjzdd ggdd g d d ggdsJdS)zTest broadcast of arguments.rrYcB @gNǣ2?rrrrr!r"rr r#N)rrr7r)r*r+r2r0r>s r8test_broadcast_argumentsz&Test_ZatValue.test_broadcast_argumentsKs tz~1v~ 6 6 !3 4        4qu XV     ) $      4qu #hZV     9 % &    rAc Ntt|jjdtjzdddsJtt|jjdtjzddgddsJt jtd 5t|jjdtjzgd dddn #1swxYwYt jtd 5t|jjdtjztj ddgdddn #1swxYwYtj ddggd gt}tt|jjdtjz|ddgdsJdS)z#`bracket` has special requirements.rN)bracketrCrrrrDsequencer%)rrDrDdtype)rrrD)rK) rrr7r)r*r+r3r4 ValueError TypeErrornparrayobject)r?rHs r8test_broadcast_bracketz$Test_ZatValue.test_broadcast_bracketlsP tz~q15y$ ? ? ?        tz~q15y1a& A A A       ]:Z 8 8 8 H H tz~q15y,,, G G G G H H H H H H H H H H H H H H H]9G 4 4 4 L L tz~q15y"(Aq6:J:J K K K K L L L L L L L L L L L L L L L(QFIII.f=== tz~q15y' B B B !3 4        s$,C  CC2>D<<EEctjtd5t|jjdt jzgdddgddddS#1swxYwYdS) zShapes mismatch as expected broadcastr%r)rr!皙?rrDrEN)r3r4rLrr7r2r*r0r>s r8test_bad_broadcastz Test_ZatValue.test_bad_broadcasts ]:[ 9 9 9    4qu "]]V                       s/AAAct|jjdtjzdd}t |tjsJ|jtj ksJ|j dksJdS)z#Test scalar input returns a scalar.rrrrEN) rr7r2r*r0 isinstanceQuantityrKrNfloat64shape)r?zs r8test_scalar_input_to_outputz)Test_ZatValue.test_scalar_input_to_outputsl  J 0$,QQ   !QZ(((((w"*$$$$w"}}}}}}rAN)__name__ __module__ __qualname__r@rFrQrUr]rWrAr8r;r;Fsa   B   @rAr;ctjtgd}tjt jd5|tjdt j zddddS#1swxYwYdS)zTest that numpy vectorize fails on Quantities. If this test starts failing then numpy vectorize can be used instead of the home-brewed vectorization. Please submit a PR making the change. )funcmethodverbose)excludedzdimensionless quantitiesr% N) rN vectorizerr3r4r*UnitConversionErrorrr)r+)rs r8test_z_at_value_numpyvectorizeris%B%B%BJ q,4N O O O-- 8<ae,,,------------------s$A//A36A3ct}t}|td|t |jdt jzd}t|j | vsJdS)NstdoutrT)rd) rrsetattrsysrr)r*r+strvaluegetvalue) monkeypatchr7 mock_stdoutresxs r8test_z_at_value_verbosertsq E**KX{333 eiQUD 9 9 9D tz??k2244 4 4 4 4 4 4rArcBrentGoldenBoundedc t}|dkrtjtd5t |jdt jz|}dddn #1swxYwY|dkrd}d }nd }d }tjtd 5tt |jdt jz|| |dsJ dddn #1swxYwYntt |jdt jz|d d dsJtt |jdt jz|d ddsJtt |jdt jz|d d dsJtt |jdt jz|d d dsJtjtd5tt |jdt jz|d d dsJtt |jdt jz|d ddsJ dddn #1swxYwYtt |jdt jz|d dd dsJtt |jdt jz|d dddsJtj tj 5tjtd5t |jdt jz|dddddn #1swxYwYddddS#1swxYwYdS)z Test 2 solutions for angular diameter distance by not constraining zmin, zmax, but setting `bracket` on the appropriate side of the turning point z. Setting zmin / zmax should override `bracket`. rxr$r%r)rcN皙?r#)g??g?)rz@zOption 'bracket' is ignored)rcrHrr)333333??)r|r()rTr{)rTr~r|)rcrHrr{)rcrHr")g333333@g@r() rr3r6rrr2r*r0 UserWarningrr4rr5)rcr7r\rHs r8test_z_at_value_bracketedrs E  \,4L M M M Y Y5:D15LQWXXXA Y Y Y Y Y Y Y Y Y Y Y Y Y Y Y s77A GGA G \+-K L L L  315L!#                         /qu "          /qu "          /qu "          /qu '         \,4L M M M  315L!&        315L!&                        * /qu "          /qu "          t* + + \,4L M M M   /qu "                    sl%AA!A8CCCA/III-M 'L<0 M<M MM MMMc&t}ddgddgddgd}|dkrtjtd}ntjtd }|5t |jd t jzd d | }dddn #1swxYwY|5t |jd t jzd d |}dddn #1swxYwYt|d||dsJt|d||d sJdS)z Test warnings on non-converged solution when setting `maxfun` to too small iteration number - only 'Bounded' returns status value and specific message. g-C6?gMbP?g{Gz?rTrurxz;Solver returned 1: Maximum number of function calls reachedr%zSolver returned Noner )rmaxfunrcN)r"rrcgŊL?rrgt)*^ @) rr3r6rrr2r*r1r)rcr7rctxz0z1s r8test_z_at_value_unconvergedr8s ED\dD\tTl S SD l O    l-5KLLL     +QYQrRX                       +QYQrRX                  Bd6l1o 6 6 6666 Bd6l1o 6 6 666666s$'BB  B 'CC  C r7c 6 d}d}tjdr|dz }tjtj}|D]_\}}|ds||vr||}t ||ddgd }t||d sJd |d`tjtrHd fd fd fdg}|D]4}||}t|t ||dd d sJ3dSdS)zn Calculate values from a known redshift, and then check that z_at_value returns the right answer. r')OkOtotangular_diameter_distance_z1z2clone is_equivalentde_density_scalewWMAP)nu_relative_density) predicate_r}r~g-q=)rHrgdy=rzRound-trip testing z failedr|c0|Sr=)_comoving_distance_z1z2rr7z2s r8z+test_z_at_value_roundtrip..su44R<<rAc0|Sr=)"_comoving_transverse_distance_z1z2rs r8rz+test_z_at_value_roundtrip..su??BGGrAc0|Sr=)rrs r8rz+test_z_at_value_roundtrip..su;;BCCrAr{)rrN) rnname startswithinspect getmembersismethodrrrX) r7r\skipmethodsrrbfvalgot func_z1z2rs ` @r8test_z_at_value_roundtriprWs6 A D 5:!!&))) (( '2BCCCG Q Q d ??3   44<< tAww tc3ZeDDDQU+++PP-P4-P-P-PPPPP%*c"" Y  < < < < < G G G G G C C C C C  Y YD477DAz$3UKKKRWXXX X X X X Y Y Y YrA)1rrmiornumpyrNr3astropyrr*astropy.cosmologyrrastropy.cosmology.funcsrrastropy.cosmology.realizationsr r r r r rrr astropy.unitsr"astropy.utils.compat.optional_depsrastropy.utils.exceptionsrmarkskipifr9r;rirt parametrizerr LambdaCDM FlatLambdaCDMwpwaCDMw0wzCDMwCDMFlatwCDMw0waCDM Flatw0waCDMrrWrAr8rs}  ((((((((BBBBBBBB                    #"""""888888777777 M*?@@&P&PA@&PR M*?@@VVVVVVVA@Vr M*?@@ - -A@ - M*?@@55A@5 M*?@@#A#A#ABByyCBA@yx M*?@@#A#A#ABB77CBA@7: M*?@@              !*0Y0Y+A@,0Y0Y0YrA