o ,e3@slddlZddlZddlZddlZddlZddlZGdddZGdddZGdddZ Gdd d Z dS) Nc@sxeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZdS) TestRemoteNhttp://raw.githubusercontent.com/dpryan79/pyBigWig/master/pyBigWigTest/test.bwcCst|j}|dus J|S)N)pyBigWigopenfnameselfbwr 1lib/python3.10/site-packages/pyBigWigTest/test.pydoOpen s  zTestRemote.doOpencCsHt|j}|dddksJWddS1swYdS)Nc Q?110)rrrchromsrr r r doOpenWiths"zTestRemote.doOpenWithcCs>|dddks J|ddksJ|ddusJdS)Nr rrrc)rrr r r doChromsszTestRemote.doChromsc Cs$|ddddddddksJdS) Nir)ZmaxValZsumDataZminValversionZ sumSquaredZnLevelsZ nBasesCovered)headerrr r r doHeader$zTestRemote.doHeadercCs|ddddgks J|jddddddgksJ|jddd dd d d d gks*J|jdtdtd dd d d d gks@J|ddgksJJdS)NrrgUUU?max)type@33?crr!nBins`ff??gq\?)statsnpint64rr r r doStatss  ,zTestRemote.doStatscCsD|dddgdks J|dtdtdgdks JdS)Nrrr)??r")valuesr*r+rr r r doValues#s*zTestRemote.doValuescCsN|ddddks J|dtdtddksJ|ddks%JdS)Nrrr)rrr-rrr.rrr")r1r2r3)dr')r5r() intervalsr*r+rr r r doIntervals(s"zTestRemote.doIntervalscCs$|jddddddddgksJdS) Nrr4r6sumrr%gA@g@B@)r)rr r r doSum-rzTestRemote.doSumcCsvtjdd}|j}|t|d}|dusJd|dfd|dfg}t|dks1J|j|dd|D];}| |d }g}g} g} g} |D]} | |d | | d | | d| | dqM|j || | | d q:|t|}| | ksJ||ksJ|D]}| |d } | |d }| |ksJq||t |dS) NFdeletewrrrrZmaxZoomsrendsr/)tempfileNamedTemporaryFilenamecloserrrlen addHeaderr7append addEntriesrosremove)rr ofileonameZbw2rrZintsZchroms2startsr@r/entryZints1Zints2r r r doWrite0s>    zTestRemote.doWritec Csdgd}gd}gd}gd}tjdd}|j}|t|d}|d d g|j|d d |d d |d d |d d d |j|d d|d d|d d|d dd |jd|dd|dddd|jd|dd|dddd|jdd|ddddd|jdd|dddddz!|j|d d |d d |d d |d d d dd ksJWn tyYnwz|jd|dd|dddddd ksJWn tyYnwz|jd|dd|dddddd ksJWn ty Ynwz|jdd|ddddddd ks$JWn ty/Ynw|jdgd |d d |d d |d d d |t t|d  }|dks_Jt |dS)z Test all three modes of storing entries. Also test to ensure that we get error messages when doing something silly This is a modified version of the writing example from libBigWig r) rr4}r$riXiqii iR)x~)?i@b@9@rZr[r\r]r^r_4r_r`rar_Fr;r=r@B2`rrr? )r/span i)r/ristepir3rerbZ ef104f198c6ce8310acc149d0377fc16N)rArBrCrDrrrFrH RuntimeErrorhashlibmd5read hexdigestrIrJ) rrrMr@r/rKrLr hr r r doWrite2UsX   22""2 " "0zTestRemote.doWrite2cCstjdd}|j}|t|d}|ddg|tt|d  }|dks0Jt|}| dd d ks@J| d dksIJ| d d dgksTJ|jd d dd dddgkscJ|t|dS)NFr;r=rbrdrpZ 361c600e5badf0b45d819552a7822937rcrf)rrerrr)r&)rArBrCrDrrrFrrrsrtrurr7r/r)rIrJ)rrKrLr rvr r r doWriteEmptys    zTestRemote.doWriteEmptyc Cstjdd}|j}|t|d}|gdtdgddgddgdd gd}tjgd tj d }tjgd tj d }tjtj d tj d }|j ||||d|ddt|||D}t|}|dddddks{JtgdD]<\} } t|| D]0\} } | d|d| | ksJ| d|d| | ksJt| d|d| | sJqq|t|dS)NFr;r=))chr1r4)chr2r5)chr3r$)chr4ryrrzr{r|)r (2<FPZZdtype)rTrk-7AKU_r?cSsg|] \}}}|||fqSr r ).0xyzr r r sz+TestRemote.doWriteNumpy..r4r5r$r})ryrzr{r|rr)rArBrCrDrrrFr*arrayr+random random_samplefloat64rHzipr enumerater7ZiscloserIrJ) rrKrLr rrMr@values0ZvalsZidx1ZchromZidx2tupr r r doWriteNumpys,  . "zTestRemote.doWriteNumpycCs|}|||jds=||||||||||| || | | | |dS)NZhttp)r rr startswithrr,r:r0r8rOrrwrxrrDrr r r testAlls         zTestRemote.testAllN)__name__ __module__ __qualname__rr rrrr,r0r8r:rOrwrxrrr r r r rs%; rc@eZdZddZdS) TestLocalcCs&t}tjtjd|_|dS)Nz/pyBigWigTest/test.bw)rrIpathdirnamer__file__rr)rZblahr r r testFoos zTestLocal.testFooN)rrrrr r r r r rc@r) TestBigBedcCstjtjd}t|}|dusJ|dksJ|dks$Jd}|}t |t r4| d}||ks:J| ddd}gd }||ksKJ| dt dt d}||ks^J|dS) Nz/pyBigWigTest/test.bigBedrratable RnaElements "BED6 + 3 scores for RNA Elements data " ( string chrom; "Reference sequence chromosome or scaffold" uint chromStart; "Start position in chromosome" uint chromEnd; "End position in chromosome" string name; "Name of item" uint score; "Normalized score from 0-1000" char[1] strand; "+ or - or . for unknown" float level; "Expression level such as RPKM or FPKM. Set to -1 for no data." float signif; "Statistical significance such as IDR. Set to -1 for no data." uint score2; "Additional measurement/count e.g. number of reads. Set to 0 for no data." ) ASCIIryii))ii(z61035 130 - 0.026 0.42 404)i7͘iQΘz61047 136 - 0.029 0.42 404)iΘisz61048 630 - 5.420 0.00 2672399)rIrrrrrZisBigWigZisBigBedSQL isinstancebytesdecodeentriesr*r+rD)rrZbbroutputoZexpectedr r r testBigBeds         zTestBigBed.testBigBedN)rrrrr r r r rrrc@seZdZddZddZdS) TestNumpyc sddl}tjdkr dSddltddjdgdddgd}jgdjd }jgd jd }jjdj d }j ||||d jgd jd }jgd jd }jjdj d }j ||||d jgdjd }jjdj d }j d|d|djgdjd }jjdj d }j d|d|djjdj d } j dddd| djjdj d } j dddd| d tddus Jfdd} | d|| d|| d|| d|| d| | d| j ddddd} t | ts?Jj ddddd} t | jsQJ |ddS)Nrz/tmp/delete.bwr=)rir>rr~) rr~rhrlrrrrrrr) rTrk#rrrrrrr?) r4nrUr5) isrQ) r$rRrSr}iiii"r)rir/) ,i6i@iJiTi^ihiri|iir)rirmr/rcsTg}d||dD] }||dq |}||dks(JdS)Nrr4rgh㈵>)r7rGrallabs)startZv2vtr r*r r compy!s   z"TestNumpy.testNumpy..compyr4r$rFnumpyT)rIrrrrFrr+rrrrHrDr/rlistZndarrayrJ) rrIrrMr@rZvalues1Zvalues2Zvalues3Zvalues4Zvalues5rZfoor rr testNumpysR  &&       zTestNumpy.testNumpyc CstjdkrdSddl}d}t|d}|j|jddddd|j|ddd|jddd s/J|j|jdddd d dd|j|dddd d |jddd sOJdS) NrrrrrhTrr)Z equal_nanZmeanrT) rrrZallcloser/rZfloat32r)r)rr*rr r r r testNumpyValues8s   zTestNumpy.testNumpyValuesN)rrrrrr r r r rs @r) rrArIsysrrrr*rrrrr r r r sJ