ó šßÈ[c@`s–ddlmZmZmZmZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZddlmZddlmZdd lmZd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#d„Z$d„Z%d„Z&d„Z'd„Z(d„Z)d„Z*d„Z+d„Z,d „Z-d!„Z.d"„Z/d#„Z0dS($i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tassert_allclosetassert_array_almost_equal_nulpi(tbiweight_locationtbiweight_scaletbiweight_midvariancetbiweight_midcovariancetbiweight_midcorrelationi(trange(tcatch_warnings(tNumpyRNGContextcC`sQtdƒ?tjjdƒ}t|ƒ}t|dƒdksGt‚WdQXdS(Ni90i'ig{®Gáz„?(R tnptrandomtrandnRtabstAssertionError(trandvartcbl((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_biweight_locations  cC`s+ttjdƒƒ}|dks't‚dS(Ni igð?(i i(RRtonesR(R((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_biweight_location_constantscC`sàd }tj|ƒ}t|ddƒ}t|tj|dƒƒt|ddƒ}t|tj|dƒƒd}d}tjdƒjddƒ}||d <||d dddg}t|ƒ}t|ƒ}t||ggƒdS(Niii(R RR(tdtcovR0((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_biweight_midcovariance_1dÒs  cC`sÏdddgdddgg}t|ƒ}d}t||| g| |ggƒdddgdddgg}t|ƒ}t|dd gd d ggƒt|d tƒ}t|dd gd d ggƒdS(Niiig<}ë`pæ?ii iôgQî©bK-@gЩ–:)ÀgŒ½´4ù®@R4gchŽ'ÆÀgáäé"¦t@(R RR5(RCRDtval((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_biweight_midcovariance_2dÙs ! cC`s5tjdƒ}t|ƒ}t|tjdƒƒdS(Nii (ii (ii(RRR RR@(RRD((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyt$test_biweight_midcovariance_constantés cC`s¶tjjdƒ}|jdddd ƒ}t|ƒ}g|D]}t|ƒ^q=}t|jƒ|ƒt|dtƒ}g|D]}t|dtƒ^q}t|jƒ|ƒdS( ua Test that biweight_midcovariance diagonal elements agree with biweight_midvariance. iiiR9idiR4N(idi( RRR:R"R RRtdiagonalR5(trngRCRDtaR0tcov2tvar2((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyt'test_biweight_midcovariance_midvarianceïs "cC`s_tjdƒjdddƒ}tjtƒ}t|ƒWdQXdt|jƒks[t ‚dS(uH Test that biweight_midcovariance raises error with a 3D array. iiNu!The input array must be 2D or 1D.( RRRtpytesttraisest ValueErrorR tstrtvalueR(RCte((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_midcovariance_shapescC`sndddg}ddgddgg}tjtƒ}t|d|ƒWdQXdt|jƒksjt‚dS(u_ Test that biweight_midcovariance raises error when M is not a scalar or 1D array. iiiitMNuM must be a scalar or 1D array.(RORPRQR RRRSR(RCRVRT((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_midcovariance_M_shape s cC`sxtjjdƒ}|jdddd ƒ}t|ƒ}t||jddƒt|dtƒ}t||jddƒd S( u| Regression test to ensure that midcovariance matrix is symmetric when ``modify_sample_size=True`` (see #5972). iiR9iiôtnulpiR4N(iiô(RRR:tgammaR RtTR5(RJRCRD((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyt%test_biweight_midcovariance_symmetrics  cC`sŽdddg}dddg}tt||ƒdƒtt||ƒdƒddddd g}d ddd d g}tt||ƒd ƒdS(Niiigð?gð¿ii gÍÌÌÌÌÌ(@gffffff*@iôgffffff@gÍÌÌÌÌÌì?g lˆ‘5r¿(RR (txR,((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyttest_biweight_midcorrelation(sc C`sñtjdƒ}tjdƒ}tjdƒ}tjtƒ/}t||ƒdt|jƒksgt‚WdQXtjtƒ/}t||ƒdt|jƒks§t‚WdQXtjtƒ/}t||ƒdt|jƒksçt‚WdQXdS(Niiiux must be a 1D array.uy must be a 1D array.u!x and y must have the same shape.(ii( RRRORPRQR RRRSR(ta1ta2ta3RT((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pyt#test_biweight_midcorrelation_inputs4s ! ! c C`s­tdƒ›tjjdƒjtjƒ}d|dR?RARBRERGRHRNRURWR[R]RaRg(((s@lib/python2.7/site-packages/astropy/stats/tests/test_biweight.pytsB"  (