U mdNO @sdZddlZddlZddlZddlmmZddl Z ddl m Z m Z ddlmZejejeZdZejedeZe eZdddgZeddgdd gged<ed d d d gddddgddddgddddgged<edddgdddgdddgged<ddZddZdd Zd!d"Z d#d$Z!d%d&Z"d'd(Z#d)d*Z$d+d,Z%d-d.Z&d/d0Z'd1d2Z(d3d4Z)d5d6Z*d7d8Z+Gd9d:d:Z,Gd;d<dd>e,Z.Gd?d@d@e,Z/GdAdBdBZ0GdCdDdDe0Z1dS)Ez' Tests for contingency table analyses. N)assert_allclose assert_equalzcontingency_table_r_results.csvresults!T~ (cCsttD]x\}}tjj|dd}|}t|jtj |dft|j tj |df|jdd}t|j|jd|j| qdS)NFZ shift_zerosZ homog_statZhomog_dfZbhapkar)methodr) enumeratetablessmstats SquareTable homogeneityr statistic r_resultslocdfsum)ktablestZhmZhmbr*h/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/tests/test_contingency_tables.pytest_homogeneity"s r,cCstjdtjtdddgd}tjddd|d<tjddd|d<t|d|d}t |}tj |}t t |}t | | t | | t|}t |dd t |jd d dS) Nidv1v2indexcolumnsrrz$A 5x5 contingency table with counts:Trrg @)nprandomseedpd DataFramerangerandintcrosstabctabr from_dataasarrayrsummaryas_textstr startswithr()r%r(rslt1rslt2Zrslt3sr*r*r+test_SquareTable_from_data/s"       rFcCsdddgdddgdddgg}tj|dddgdddgd}tj|dd }tjddddgddddgddddgddddggtjd }t||jdS) Nrrrr rrr0FrZdtype) r7r8r<r r4r>float64rr()tabr%Zdf2er*r*r+test_SquareTable_nonsquareFs.rKcCsdddgdddgddd gg}t|}t|}|j}t|d d td t|d dtddddtt||jddddS)Nrr rrrr r37rr-Hh㈵>atolrtol) r4r>r<TableZcumulative_oddsratiosrfloatlogZcumulative_log_oddsratios)r(tbl_objZcum_oddsr*r*r+test_cumulative_oddsSs  r\cCs|dddgdddgddd gg}t|}t|}|j}t|d d t|d d tddddtt||jddddS)Nrr rrrrrLrMrNr3g?rQ rrTrU) r4r>r<rXZlocal_oddsratiosrrYrZZlocal_log_oddsratios)r(r[Zloc_oddsr*r*r+test_local_oddsas  r^cCstjdtjd}tdd}ttj|ddj|ttj|ddj|tjddd gd dd gd d dggtjd}tjddd gd dd gd d dggtjd}tt|j|ttj|ddj|dS)N)rrrGg?FrTrrr rrrr) r4zerosrHfullrr<rXr(r>)tresultrr*r*r+ test_shiftingos$   rdcCsnddgddggddgddggddgddggg}t|j}t|}t|}t|j|jt|j|jdS)NrMrNrrLrr )r4r>Tr<StratifiedTableroddsratio_pooledlogodds_pooled)tab1Ztab2Zct1Zct2r*r*r+test_stratified_table_cubes .   rkcCsddddgddddgg}d d d d gd dddgg}ddddgddddgg}tddddgdddd gg}td!d"d#d$gd%d&d'd(gg}t|}t|j|d)d)d*t|j|d)d)d*t|j|d)d)d*t|j|d)d)d*dS)+Nr]rMr)i3igh|?6@gh|?1@gS㥛D@@g?5^I3@gQr@g= ףpm@g(\W{@g33333p@g|a2U0j@g1Z@g(\?gNbX96@g'?gw/?g_vOv?gho?g|ns;*g=yHg u5;ѿg}6`F@g8~?g]:?gW̲?g6gU朎hg-ջ.gW!<7ֿg0L@gU朎h@g-ջ.@gW!<7?g0L-C6?rU) r4arrayr<rXrZ fittedvaluesZ chi2_contribsZ resid_pearsonZstandardized_resids)r(fitc2prsrrIr*r*r+ test_residss$       rucCsnttD]^\}}dt|jd}dt|jd}tj|dd||}t|j t j |dft|j t j |dft|j dt j |dft|jdt j |d fd d d t|jt j |d fd d d tj|dd||d}t|j t j |d ft|j t j |dft|j dt j |dft|jdt j |dft|jt j |dfd d d qdS)NrrFrZlbl_statZ lbl_expvalr Zlbl_varZlbl_chi2rTrWrVZ lbl_pvalueZ lbl2_statZ lbl2_expvalZlbl2_varZ lbl2_chi2Z lbl2_pvalue)rrr4arangeshaper<rXtest_ordinal_associationrr"r#r$Z null_meanZnull_sdZzscorepvalue)r'r(Z row_scoresZ col_scoresrsltr*r*r+rys rycCsdtjdtjjdddd}ddlm}||}t|}t |j |dt |j |ddS) Ni'"rr)rr)sizer)chi2_contingencyr) r4r5r6r:Z scipy.statsr}r<rXZtest_nominal_associationrr"rz)r(r}Z rslt_scipybr*r*r+test_chi2_associations  rcCspttD]b\}}tjj|dd}|}t|jtj |dft |j tj |dft|j tj |dfqdS)NFrZ bowker_statZ bowker_dfZ bowker_pvalue) rrrrr Zsymmetryrr"r#r$rr%rz)r'r(r)r~r*r*r+ test_symmetrys rcCstjtdddd}tjtd}|}t|j|jt |j dtjtdddd}t|j t j dtjtddd}t|j t j ddS) NrFexact correctionrT)rZ homog_cont_p)r)rZ homog_binom_p)r<mcnemarrrrr r!rr"rr%rzr#r$)b1r)b2Zb3Zb4r*r*r+ test_mcnemars rc Cstddddddddgddddddddgddddddddggj}tddgddggddgddggg}tjddd|}t|j |tjdddt|}t|j |dS)Nrrr ) r7r8rfr4r>r<rgr=rr(r%rJrjr*r*r+test_from_data_stratifieds(& rc Cstddddddddgddddddddggj}tddgddgg}tjj|dd}t|j |tjjt|dd}t|j |dS)Nrrr rFr) r7r8rfr4r>r<Table2x2r=rr(rr*r*r+test_from_data_2x2s 0 rc Csddddgddddgddddgddddgddddgddddgddddgddddgddddgg }t|}tj|dd\}}}t|dt|ddddddgdddddgdddddgdddddgdddddgdddddgdddddgdddddgdddddgg }t|}tj|dd\}}}t|ddd t|d |dddd f}tt|dddf|dddf}tj|d d}tj|ddd }t|j|jt|j |j t t | dd dS)NrrF)Z return_objectg@rg h"lxz?ro)rWrr Trzdf 1 pvalue 0.65) r4r>r<Z cochrans_qrr7r;rr"rzrrArB)r(statrzr%dataZxtabrrr*r*r+ test_cochranqsF                       *rc@sZeZdZedddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)CheckStratifiedMixinFcCsL|s|nt|}t||_tj|dd|_dd|D}t||_dS)NTrcSsg|]}t|qSr*)r7r8).0xr*r*r+ Msz3CheckStratifiedMixin.initialize..)r4Zdstackr<rgr{Zrslt_0 rslt_pandas)clsruse_arrZtables1Z tables_pandasr*r*r+ initializeHs  zCheckStratifiedMixin.initializecCst|jj|jddddSNrorv)rr{rhselfr*r*r+test_oddsratio_pooledQs z*CheckStratifiedMixin.test_oddsratio_pooledcCst|jj|jddddSr)rr{rirr*r*r+test_logodds_pooledVs z(CheckStratifiedMixin.test_logodds_pooledcCs:|jjdd}t|j|jdddt|j|jddddS)NT)rrorTrv)r{test_null_oddsrr"mh_statrz mh_pvaluerr{r*r*r+r[sz#CheckStratifiedMixin.test_null_oddscCs6|j\}}t||jdddt||jddddSr)r{Zoddsratio_pooled_confintror_lcbor_ucbrZlcbZucbr*r*r+test_oddsratio_pooled_confintasz2CheckStratifiedMixin.test_oddsratio_pooled_confintcCsB|j\}}t|t|jdddt|t|jddddSr)r{Zlogodds_pooled_confintrr4rZrrrr*r*r+test_logodds_pooled_confintgsz0CheckStratifiedMixin.test_logodds_pooled_confintcCs~t|dsdS|jjdd}t|j|jdddt|j|jddd|jjdd}t|j|jdddt|j|j ddddS)Nor_homogF)adjustrorvT) hasattrr{test_equal_oddsrr"rrz or_homog_p or_homog_adjor_homog_adj_prr*r*r+ros z$CheckStratifiedMixin.test_equal_oddsc CsBt0tdtt|j|jW5QRXdS)Nignore) warningscatch_warnings simplefilterRuntimeWarningrr{r?r@rrr*r*r+ test_pandas}s    z CheckStratifiedMixin.test_pandasc Cstjdtjtddd}tjddd|d<tjddd|d<ttd t d |d <g}td D]B}td |d |d }| t |j |df|j |dfqnt |}t jddd |}t||dS) Nr-)r.r/stratr0rr r.r/rrr)r4r5r6r7r8r9r:ZkronrwZonesappendr;r$r<rgr=rr?r@)rr%rr'iirCrDr*r*r+test_from_datas  ( z#CheckStratifiedMixin.test_from_dataN)F) __name__ __module__ __qualname__ classmethodrrrrrrrrrr*r*r*r+rFs rc@seZdZdZeddZdS)TestStratified1z data = array(c(0, 0, 6, 5, 3, 0, 3, 6, 6, 2, 0, 4, 5, 6, 1, 0, 2, 5, 0, 0), dim=c(2, 2, 5)) rslt = mantelhaen.test(data) cCsdgd}tddgddgg|d<tddgddgg|d<tddgddgg|d<tddgddgg|d<tddgddgg|d<||d|_td|_d |_d |_d |_d |_ dS) Nrrrrrr rrLgY8m@g M?g| Vjm?g |(G@) r4rprrhrZrirrrrrrr*r*r+ setup_classs   zTestStratified1.setup_classNrrr__doc__rrr*r*r*r+rs rc@seZdZdZeddZdS)TestStratified2a_ library(DescTools) data = array(c(20, 14, 10, 24, 15, 12, 3, 15, 3, 2, 3, 2, 12, 3, 7, 5, 1, 0, 3, 2), dim=c(2, 2, 5)) rslt = mantelhaen.test(data) bd1 = BreslowDayTest(data, correct=FALSE) bd2 = BreslowDayTest(data, correct=TRUE) cCsdgd}tddgddgg|d<tddgd dgg|d <td d gd d gg|d <tdd gd dgg|d <td dgd d gg|d <|j|ddd|_td|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ dS)Nrrrrrrr]rrr rLrT)rg_)Ǻ @g48'@gq?FB?g,i?g}ZEh@g9m4?gX9v?gV}b?r4rprrhrZrirrrrrrrrrr*r*r+rs"  zTestStratified2.setup_classNrr*r*r*r+rs rc@seZdZdZeddZdS)TestStratified3a library(DescTools) data = array(c(313, 512, 19, 89, 207, 353, 8, 17, 205, 120, 391, 202, 278, 139, 244, 131, 138, 53, 299, 94, 351, 22, 317, 24), dim=c(2, 2, 6)) rslt = mantelhaen.test(data) bd1 = BreslowDayTest(data, correct=FALSE) bd2 = BreslowDayTest(data, correct=TRUE) cCsdgd}tddgddgg|d<tddgd d gg|d <td d gddgg|d<tddgddgg|d<tddgddgg|d<tddgddgg|d<||d |_td |_d!|_d"|_d#|_d$|_ d%|_ d&|_ d%|_ d&|_ dS)'Nri9irYriarMrxir ir5i+^ri_i=rrgcK?gOec?g6>W[?gZ ?gA%Pi?g=2@g+`?rrr*r*r+rs$   zTestStratified3.setup_classNrr*r*r*r+rsrc@seZdZeddZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZdS) Check2x2MixincCs"t|j|_tj|j|_dSN)r<rr(r[r=r tbl_data_obj)rr*r*r+rszCheck2x2Mixin.initializecCst|jj|jdSr)rr[ oddsratiorr*r*r+test_oddsratioszCheck2x2Mixin.test_oddsratiocCst|jj|jdSr)rr[ log_oddsratiorr*r*r+test_log_oddsratiosz Check2x2Mixin.test_log_oddsratiocCst|jj|jdSr)rr[log_oddsratio_serr*r*r+test_log_oddsratio_sesz#Check2x2Mixin.test_log_oddsratio_secCst|j|jdSr)rr[oddsratio_pvaluerr*r*r+test_oddsratio_pvalue sz#Check2x2Mixin.test_oddsratio_pvaluecCs2|jd\}}|j\}}t||t||dSNg?)r[oddsratio_confintrrZlcb1Zucb1Zlcb2Zucb2r*r*r+test_oddsratio_confint$s  z$Check2x2Mixin.test_oddsratio_confintcCst|jj|jdSr)rr[ riskratiorr*r*r+test_riskratio+szCheck2x2Mixin.test_riskratiocCst|jj|jdSr)rr[ log_riskratiorr*r*r+test_log_riskratio/sz Check2x2Mixin.test_log_riskratiocCst|jj|jdSr)rr[log_riskratio_serr*r*r+test_log_riskratio_se3sz#Check2x2Mixin.test_log_riskratio_secCst|j|jdSr)rr[riskratio_pvaluerr*r*r+test_riskratio_pvalue7sz#Check2x2Mixin.test_riskratio_pvaluecCs2|jd\}}|j\}}t||t||dSr)r[riskratio_confintrrr*r*r+test_riskratio_confint;s  z$Check2x2Mixin.test_riskratio_confintcCs2|jd\}}|j\}}t||t||dSr)r[log_riskratio_confintrrr*r*r+test_log_riskratio_confintBs  z(Check2x2Mixin.test_log_riskratio_confintcCs"t|j|jdSr)rr[r?r@rrr*r*r+rIs zCheck2x2Mixin.test_from_datacCst|j|jdSr)rr[r?r@summary_stringrr*r*r+ test_summaryMszCheck2x2Mixin.test_summaryN)rrrrrrrrrrrrrrrrrrr*r*r*r+rs rc@seZdZeddZdS) Test2x2_1cCstd}ddddddddg|dddf<ddddddddg|dddf<t||_tddgddgg|_d|_d|_td|_ddg|_ d|_ d|_ d|_ dtd|_ d|_d d g|_d d g|_d ddddddg}d||_|dS)N)rMr rrr g?ggU3?g~3/@gvc?gL%@g,},g,},?z3 Estimate SE LCB UCB p-valuez3---------------------------------------------------z3Odds ratio 1.000 0.063 15.988 1.000z3Log odds ratio 0.000 1.414 -2.772 2.772 1.000z3Risk ratio 1.000 0.250 3.998 1.000z3Log risk ratio 0.000 0.707 -1.386 1.386 1.000 )r4r_r>rr(rrsqrtrrrrrrrrrjoinrr)rrssr*r*r+rUs<       zTest2x2_1.setup_classN)rrrrrr*r*r*r+rSsr)2rosrnumpyr4Z$statsmodels.stats.contingency_tablesrZcontingency_tablesr<Zpandasr7Z numpy.testingrrZstatsmodels.apiapirpathdirnameabspath__file__Zcur_dirfnamerZfpathZread_csvr#rr>r,rFrKr\r^rdrkruryrrrrrrrrrrrrr*r*r*r+sV           AR,,E