U e{jM4@sddlZddlZddlmZddlZddlZddlmZ ddl m Z m Z ddl mZmZddlmZmZmZmZmZmZmZmZddlmZddlmZdd lmZejj e d d d d Z!ejj e d d GdddZ"ejj e d d ddZ#ejj e d d ddZ$ejj e d d ej%ddddgddZ&ejj e d d ej%ddddgddZ'ejj e d d ej%deeeeeeeee j(e j)e j*e j+e j,e j-e j.e j/gddZ0dS)N)StringIO)units)coreflrw)_z_at_scalar_value z_at_value)WMAP1WMAP3WMAP5WMAP7WMAP9Planck13Planck15Planck18)allclose) HAS_SCIPY)AstropyUserWarningztest requires scipy)reasonc Cst}tt|jdtjddds$ttt|jdtjdddsDttt|jdtj dddsdttt|j d tj d ddsttt|j d tj d d dddsttt|j dtj dddddsttt|j dtj dddddstttj6tjtddt|j dtj ddW5QRXW5QRXttj6tjtddt|j dtj ddW5QRXW5QRXdS)Ng&W @ư>ZrtolgCr?.gʲb?g@g=*?g9{ :@g|=)ztolg& .>)zmax,?@)zminԭ T@fval is not bracketedmatch?@)r rrageuGyrAssertionErrorZ lookback_timeZdistmodmagZluminosity_distanceMpcGpcangular_diameter_distancepytestraisesrCosmologyErrorwarnsr)cosmor3g/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/cosmology/funcs/tests/test_funcs.pytest_z_at_value_scalars<       *r5c@s4eZdZddZddZddZddZd d Zd S) Test_ZatValuecCs t|_dSN)r r2selfr3r3r4 setup_classHszTest_ZatValue.setup_classcCstt|jjddgtjddgdds*ttt|jjdtjdd gdd gd d d gdds^ttt|jjdtjdd ggdd gd d d ggddstdS)zTest broadcast of arguments.rrYcB @gNǣ2?rrrrrrrrr N) rrr2r&r'r(r)r-r+r8r3r3r4test_broadcast_argumentsKs2  z&Test_ZatValue.test_broadcast_argumentsc Cstt|jjdtjddddds&ttt|jjdtjddgddddsPttjt d d &t|jjdtjdddd gdW5QRXtjt d d (t|jjdtjt ddgdW5QRXt j ddgdd dggt d}tt|jjdtj|dddgddstdS)z#`bracket` has special requirements.rN)bracketr;rrrr<sequencer"dtyper)rB)rrr2r&r'r(r)r.r/ ValueError TypeErrornparrayobject)r9r?r3r3r4test_broadcast_bracketls(  *,z$Test_ZatValue.test_broadcast_bracketc CsBtjtdd*t|jjdtjdddgddgd W5QRXd S) zShapes mismatch as expected broadcastr"rrr皙?rr<r=N)r.r/rCrr2r-r'r+r8r3r3r4test_bad_broadcastsz Test_ZatValue.test_bad_broadcastcCsLt|jjdtjddd}t|tjs*t|jt j ks:t|j dksHtdS)z#Test scalar input returns a scalar.rrrr=r3N) rr2r-r'r+ isinstanceZQuantityr)rBrEZfloat64shape)r9zr3r3r4test_scalar_input_to_outputsz)Test_ZatValue.test_scalar_input_to_outputN)__name__ __module__ __qualname__r:r>rHrKrOr3r3r3r4r6Fs !  r6c CsFtjtdddgd}tjtjdd|tjdtj W5QRXdS) 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) rEZ vectorizerr.r/r'ZUnitConversionErrorrr&r()rr3r3r4test_z_at_value_numpyvectorizes rXcCsHt}t}|td|t|jdtjdd}t|j | ksDt dS)NstdoutrT)rU) r rsetattrsysrr&r'r(strvaluegetvaluer))Z monkeypatchr2Z mock_stdoutZresxr3r3r4test_z_at_value_verboses r_rTBrentGoldenBoundedc CsDt}|dkrtjtddt|jdtj|d}W5QRX|dkrNd}d}nd }d }tjtd d,t t|jdtj||d |d dst W5QRXnZt t|jdtj|dd d d dst t t|jdtj|dd dd dst t t|jdtj|dd d d dst t t|jdtj|dd d d ds6t tjtddVt t|jdtj|dd d d dsnt t t|jdtj|d d dd dst W5QRXt t|jdtj|d ddd d dst t t|jdtj|ddddd dst t t j :tjtdd t|jdtj|dddW5QRXW5QRXdS)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`. rbr!r"r)rT皙?r )g??g?)rc@zOption 'bracket' is ignored)rTr?rr)333333??)rer%)rJrd)rJrgre)rTr?rrd)rTr?r)g333333@g@r%N)r r.r1rrr-r'r+ UserWarningrr)r/rr0)rTr2rNr?r3r3r4test_z_at_value_bracketeds          ric Cst}ddgddgddgd}|dkr4tjtdd}ntjtd d}| t|jd tjd d |d }W5QRX| t|jd tjd d |d}W5QRXt|d||ddst t|d||d dst dS)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?rJ)r`rarbrbz;Solver returned 1: Maximum number of function calls reachedr"zSolver returned Noner )rmaxfunrT)rrlrTgŊL?rrgt)*^ @N) rr.r1rrr-r'r,rr))rTr2rctxZz0z1r3r3r4test_z_at_value_unconverged8s4ror2c sd}d}tjdr |d7}tjtjd}|D]V\}}|ds4||krPq4||}t||ddgd d }t||d d s4td |dq4t jtrdfddfddfddg}|D]*}||}t|t||dd dd d stqdS)zn Calculate values from a known redshift, and then check that z_at_value returns the right answer. r$)ZOkZOtotangular_diameter_distance_z1z2cloneZ is_equivalentZde_density_scalewZWMAP)Znu_relative_density) predicate_rfrgg-q=)r?rgdy=rzRound-trip testing z failedrecs |Sr7)Z_comoving_distance_z1z2rnr2Zz2r3r4z+test_z_at_value_roundtrip..cs |Sr7)Z"_comoving_transverse_distance_z1z2rurvr3r4rwrxcs |Sr7)rprurvr3r4rwrxrd)rrN) r\name startswithinspect getmembersismethodrrr)rL) r2rNskipmethodsryrSZfvalgotZ func_z1z2r3rvr4test_z_at_value_roundtripWs(       r)1r{r[iornumpyrEr.Zastropyrr'Zastropy.cosmologyrrZastropy.cosmology.funcsrrZastropy.cosmology.realizationsrr r r r r rrZ astropy.unitsrZ"astropy.utils.compat.optional_depsrZastropy.utils.exceptionsrmarkZskipifr5r6rXr_Z parametrizeriroZ LambdaCDMZ FlatLambdaCDMZwpwaCDMZw0wzCDMZwCDMZFlatwCDMZw0waCDMZ Flatw0waCDMrr3r3r3r4s^  (    )Y   |