L ixddlZddlmZmZddlZddlmZm Z m Z ddl m Z ddl mZGddZGdd Zy) N)raiseswarns)assert_approx_equalassert_allclose assert_equal)cdist)statsceZdZdZdZdZdZdZdZe jjddd gd Z d Z d Zy )TestMGCErrorWarningsz1 Tests errors and warnings derived from MGC. ctjd}dgdz}tttj ||tttj ||y)Nnparange assert_raises ValueErrorr multiscale_graphcorrselfxys `/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/stats/tests/test_mgc.pytest_error_notndarrayz*TestMGCErrorWarnings.test_error_notndarray sB IIbM C"Hj%"<">?A '   AUAJ 7A   AUAJ 7A))$$Q#UAJ$?C))$$Q#UAJ$?C!ae aA!ae aA,- - !8))!QeT!V_)EJ:Q7A!t rzsim_type, obs_stat, obs_pvalue))rG ףp= ?MbP?)rKw/?rf)rLgUN@g(\?ctjjd|jdd|\}}t j ||\}}}t ||dt ||dy)NNarrHr[r\r] significantrrQseedrdr rr rr]obs_stat obs_pvaluerrstatpvalue_s r test_onedzTestMGCStat.test_onedksc x   sX F1 44Q:faD(:FJA>r)rGgZd;O?rf)rKg~jt?gʡE?ctjjd|jdd|\}}t j ||\}}}t ||dt ||dy)NrirrrjrHrkrmros r test_fivedzTestMGCStat.test_fived|sc x   sX F1 44Q:faD(:FJA>rctjjdtjjddd}tjj ddd}t j ||\}}}t|d d t|d d tjj ddd}t j ||d \}}}t|d d t|d d y)NrirrM)rrrIrrH)Prg?rkrfT) is_twosamp)rrQrnrYrXr rrrrrrrrsrts r test_twosampzTestMGCStat.test_twosamps x  II  sCh  7 II  Q  0 44Q:faD#15FEq9 II  Q  1 44QdKfaD#15FEq9rctjjd|jddd\}}t j ||d\}}}t |dd t |d d y) NrirrHrGrjrO)workersrerkrfrmr{s r test_workerszTestMGCStat.test_workerssc x   sX F1 44Q1EfaD$A6FEq9rc|jddd\}}tj||d\}}}t|ddt|ddy) NrrHrGrj random_statererkrf)rdr rrr{s rtest_random_statezTestMGCStat.test_random_statesS  sX F1 44QJfaD$A6FEq9rctjjd|jddd\}}t ||d}t ||d}t j ||dd\}}}t|d d t|d d y) NrirrHrKrj euclidean)metric)r,rrgrkrf)rrQrnrdrr rr)rrrdistxdisty stat_dist pvalue_distrts rtest_dist_permzTestMGCStat.test_dist_perms x   s[ I1a;/a;/$)$>$>uePTLM%O! ; Iu!<KA>rr ctjjd|jddd\}}t j ||d\}}}t |dy)NrirrHrGrjrg'^P?)rrQrnrdr rr)rrrrtrss rtest_pvalue_literaturez"TestMGCStat.test_pvalue_literaturesW x   sX F111!QQG 61'rctjjd|jddd\}}t j ||d}t |j|jy)NrirrHrGrjr) rrQrnrdr rrrr statistic)rrrress r test_aliaszTestMGCStat.test_aliassU x   sX F1((AA>SXXs}}-rN)rrH)r<r=r>r?rdr@rAxslowrBrurwr|rrr fail_slowslowrrrCrrrErEFsZ D [[ [[=@ ?  ? [[ [[=!$@ ?  ? [[::( [[ : : [[:: [[ ? ? [[2 [[(( [[..rrE)r@rrrr5numpyr numpy.testingrrrscipy.spatial.distancerscipyr r rErCrrrs3 ALL(9D9DxS.S.r