U e{j @sddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z mZmZddZd d Zejd d efdefd d ddgeeeeffddddgeeeeffgddZGdddZdS))infN)aszarrinf_likevectorize_if_neededvectorize_redshift_method)AstropyDeprecationWarning)_zarr invalid_zsvalid_zscCs(Gddd}|}t|jds$tt|jjtjs8t|ddksJtt|dts^t|tdtdks|tt|tdtjstt |t ddgt ddgkstt|t ddgtj stt |ddgt ddgkstt|ddgtj s$tdS)z?Test :func:`astropy.cosmology.utils.vectorize_redshift_method`.c@seZdZeddZdS)z-test_vectorize_redshift_method..ClasscSs|S)N)selfzr r a/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/cosmology/tests/test_utils.pymethodsz4test_vectorize_redshift_method..Class.methodN)__name__ __module__ __qualname__rrr r r rClasssr__vectorized__rN) hasattrrAssertionError isinstancernpZ vectorizeintfloat64allarrayZndarray)rcr r rtest_vectorize_redshift_methods* &r c CsRdd}tt6t|ddks&ttt|ddgddgksDtW5QRXdS)z Test :func:`astropy.cosmology.utils.vectorize_if_needed`. There's no need to test 'veckw' because that is directly pasased to `numpy.vectorize` which thoroughly tests the various inputs. cSs|dS)Nrr )xr r rfunc9sz&test_vectorize_if_needed..funcr N)pytestwarnsrrrr)r"r r rtest_vectorize_if_needed1s r(z arr, expectedgg?@@rr$c Cs0tttt||ks"tW5QRXdS)z Test :func:`astropy.cosmology.utils.inf_like`. All inputs should give a float output. These tests are also in the docstring, but it's better to have them also in one consolidated location. N)r&r'rrrrr)Zarrexpectedr r r test_inf_likeCs r,c@s\eZdZejdeeeddde ddde e e e g dd Z ejd e d d Zd S) Test_aszarrz z, expectrriLi r)r*cCst|}t||stdSz,Test :func:`astropy.cosmology.utils.aszarr`.N)rrZ array_equalr)r rexpectgotr r r test_valid[s zTest_aszarr.test_validzz, excc Cs"t|t|W5QRXdSr.)r&Zraisesr)r rexcr r r test_invalidis zTest_aszarr.test_invalidN)rrrr&mark parametrizelistzipr rrr r1r r3r r r rr-Zs  r-)mathrnumpyrr&Zastropy.cosmology.utilsrrrrZastropy.utils.exceptionsrZ test_corer r r r r(r4r5r,r-r r r rs"