K iDddlmZddlZddlZddlmZmZmZdZ dZ dZ ejjddd Zd Zd Zd Zd ZdZdZGddZy))chainN)assert_array_equal assert_equal assert_raisesc gdgdggdgdgg}dD]}}tj||}tj|d}t|jtj t |tjd gd ggd gd gggtttjtj|ty) N)rr)rrr)rrr)rrr ?bBhHiIlLqQdtypeaxis@ ) nparraypackbitsrr uint8rr TypeErrorfloat)adtarrbs c/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/lib/tests/test_packbits.py test_packbitsr s Y  Y  !AHhhq# KK" %QWWbhh'1bhh# t}'EFG H )R[["((1E*BCcgd}dD]o}|D]h}tj||}tj|}t|jtj t|j djqy)N)r rr#rr$rr#r$r$rrrr$rrrr$rrrr r r!)remptyrrr rshape)shapesrr,rrs rtest_packbits_emptyr.sfF( (Eb)A AA "(( + $ '  ((rc nddgfdgdfdgdfdgdfdgd fd gd fd gd fdgdfg}dD]}|D]\}}t|D]m\}}tj||}tj||}t |j tj t |j|oy)Nr!r"))r$r)r#rr"r%))r0rr$r%)r#rr1r&)r&)rr0r$)rr#r1r')r1rrr'r'r()r(rr1rr(r))r)r)rrr1r*r*r*r*r r r ) enumeraterr+rrr rr,)r-rin_shape out_shapesax out_shaperrs rtest_packbits_empty_with_axisr;#s v ;< ;< ;< 89 89 89 56 F1$* 1 Hj!*:!6 1 IHHXR0KK+QWWbhh/QWWi0  1 11rbitorderlittlebigctjgd}|jd}dD]}tj||}tj|d|}t |j tj gd}|dk(r t||ttj||dd |td d Dcgc] }tj|d| d d "}}t|gd|jdd}tj|d }t |j tj t|gdgdgdgdgdgtj|d }t |j tj t|ggdgdgdgdgdgdgdgdgdgd gd!gd"gd#gd$gd%gd&gd'gd(gd)gd*gdgd+gd,gd-gd.gd/gd0gd1gd2gd3gd4gd5gd6gd7gd8gd9|jj}tj|d }t |j tj t|gd:gd;gd<gd=gtj|d }t |j tj t|gd>gd?gd@gdAgdBgdCgdDgdEgdFgdGgdHgdIgdJgdKgdLgdMgdNgdOgdPgdQgdRgdSgdTgdUgdVgdWD]}tj||}tjjtj|j tj|j"|j$|X}d ||dk(<||j'|z}tj|d }ttj|dd |t)t*tjtj|t,ycc}w)YN(,rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr1r r rr<)qrr1rBrrErrrBqr1rIrrHrrH~rrCrIrErFrP?rHrM8rOrGrrOrQrNrRrBprRrIrKr1rGrRrNrUrQrIrIrNrTrrFrr1rIrWrrrIrNrrIrBrMrQrrrrBrTrQrrLrWrBrJrIr1rSrRrBrTrErrJrIrIrPrRrOrKrOrYrWrFrSr?r<rr r )rSrSrSrFrHr\rrrrWr$r)reWSrirWrlrmrn(ikKJX)HrKrVrOrjZryririwrCmIr-)hzry)rJxrKrwr^<4rfrfrfrrIrIr)rrrDrrrvru\rNn'rrx)rrSrSrPrU0rrrNrGr}rSrrNrGr}rrrGrNrrS)rBrCrr)rErBrLrS)rGrrHr)rIrSrrS)rrFrIrS)rPrQrr)rIrGrEr)rErNrTr)rMrrNr)rIrIrWr)rRrHrMr)rJrKrOrS)rOrPrQr)rrHrUr)rLrKr1rS)rHrMrRr)rRrIrVrS)rGrErNr)rOrYrrS)rIrIrDr)r1rVrKr)rrIrVrS)rSrrIrS)rNrrIrS)rKrBrMr)rrEr1rS)rNrJrKr)rrBrCrS)rPrQrNr)rNrTrQr)rEr1rSr)rJrIrIrS)rHrJrWr)rErOrPr)rTrRrBr)$rBrErGrIrrPrIrErMrIrRrJrOrrLrHrRrGrOrIrMr1rrSrNrKrrNrrPrNrErJrHrErT)$rCrBrrSrFrQrGrNrrIrHrKrPrHrKrMrIrErYrIrrVrIrrrBrErJrBrQrTr1rIrJrOrR)$rrLrHrrIrrErTrNrWrMrOrQrUr1rRrVrNrrDrNrKrVrIrIrMr1rKrCrNrQrSrIrWrPrB)$rrSrrSrSrrrrrrrSrrrSrrSrrSrrrrSrSrSrrSrrSrrrrSrrr)rcrvrJrr)rdrwrrrS)rerKrKrrS)rerVrwrr)rfrOrrr)rgrjr^rrU)rhrxrrr)riryrrr)riryrrvr)rjrirfrurN)rWrirfrrG)rkrzrfrr})rlrCrrr)rlr{rrrS)rmr|rrr)rmrrrrN)rnr}rIrrG)rnr~rIrr})rorrrr)rprrrr)rqrrrr)rrrrrxrG)rsrrrrN)rtryrrr)rurrrrS bBhHiIlLqQ)lowhighsizer )rrrepeatrrr rr unpackbitsrangereshapeTcopyrandomrandintiinfominmaxrastyperrr)r<rr rrrirnds rtest_packbits_larger7s N OA  A};hhq& KK$ :QWWbhh' / u  q! $2==X>sCQG? @ KK! $QWWbhh'1#53#53#53#5 4#5 4 #5 4 #5 4 #5 4#5 4#5 4#5 4#5 4#5 4#5 4#5 4#5 4#5 4!#5" 4##5$ 4%#5& 4'#5( 4)#5* 4+#5, 4-#5. 4/#50 41#52 43#54 45#56 47#58 49#5: 4;#5< 4=#5> 4?#5@ 4A#5B 4C#5D 4E#5F 4G#5# 6Jeejjl KK! $QWWbhh'1 3!7!:!MN O KK! $QWWbhh'188888888888888888888888881: ;K};@5hhq&iiBHHUO$7$7%'XXe_%8%8sxx&+ -C1H  szz%   KK" %2==+CR0!45)R[["((1E*BCq Is %QctddD]=}dD]6}tjd|ft}tj|d8?y)Niir r rr )rronesboolr)srxs rtest_packbits_very_largersK3 # #Ba-A KK " ##rc  tjdgdgdggtj}tj|d}t |j tjt |tjgdgdgd gy) Nr0rEr rr )rrrrrrrr)rrrrrrrr)rrrrrrrr)rrrrrr rrrs rtest_unpackbitsrsi 1#sRD!2A aa A"((#q"(($<$<$<$>?@rctjdgdgdggtj}tj|d}t |j tjtj|dd}tj|dd }t ||t |tj|ddt |ddddd f|t |tj|dd tttj|d tttj|d y)Nr0rErr rr r>rAr?r rrZr#) rrrrrr rrr ValueErrorr)rrb_littleb_bigs rtest_pack_unpack_orderrs 1#sRD!2A aa A"((#}}QQ:H MM!!e 4Eq% q"++hQJKqDbDz8,q"++e!eDE*bmmQ=)R]]A;rctjdtj}tj|}t |j tjt |tjdy)Nr!r )rr+rrrr rrs rtest_unpackbits_emptyrsH RXX&A aA"((#q"((4.)rc |dgdfgddfgddfgddfgdd fgd d fgd d fgddfg}|D]\}}t|D]{\}}tj|tj}tj||}t |j tjt |j|}y)Nr!))r0r^r)r\r1rr\r^rr))r0rr^r\rr^)r\rr1r)rr\r^)rr0r^)rr\r1r)r2r^rrrr)rr^rr3rr)rrr^rr4rr5r*r r )r6rr+rrrr r,)r- in_shapesr:r9r7rrs rtest_unpackbits_empty_with_axisrs  . < . < . < ,j9 ,j9 ,j9 *I6 F!'- 9%i0 -LB2A ab)A "(( + ) ,  --rc ftjdtj}ttjtj ||ttjtj |ddd|dddtj |d}ttjtj |dd||jj}ttjtj |dd|y)Nir r0)r1rrr r) rarangerrrrtilerr)ds rtest_unpackbits_largers #RXX&Ar{{2==#34a8r{{2==3Q3#891SqS6B 6Ar{{2==#;!DaH  Ar{{2==#;!DaHrc eZdZejgdgdgdgdgdgdgdgej Zejd ej Zejed d ejd ej Z ed d d jje d d ejd ej Z ee d dd df<e jjdde jjdeeded dd dZe jjdidd igdZe jjdde jjdeeded dd dZe jjdidd iddidd dddidd dgdZdZy ) TestCount)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrr)rrrrrrrr 9N1r ) rrEr<r=count:c |dkr|dz }n|}tj|j|}tj|||}t |j tj t||jd|y)NrrrZ)rr< rrrrrr rrpadded1)selfr<rcutoffpackedunpackeds rtest_roundtripzTestCount.test_roundtrip.sf 19QYFFTVVh7==uxHX^^RXX.8T\\'6%:;rkwargsctj|j}tj|fi|}t |j tj t||jddy)Nr r)rrrrs r test_countzTestCount.test_count<sPTVV$==262X^^RXX.8T\\#2%67rrocr|dkr|dz }n|}tj|jd|}tj|d||}t |j tj t||jd|d|jjdftj|jd|}tj|d||}t |j tj t||jd|jjdd|fy)NrrrA)rrr<) rrrrrr rrpadded2r,)rr<rrpacked0 unpacked0packed1 unpacked1s rtest_roundtrip_axiszTestCount.test_roundtrip_axisEs 19QYFF++dff1x@MM'+35 Y__bhh/9dll7F7)r<rr?ctj|jd}tj|fddi|}t |j tj |jdddk(r7t||jddd|jjdfn@t|dddddf|jddd|jjdftj|jd}tj|fddi|}t |j tj |jdddk(r7t||jd|jjdddfyt|dddddf|jd|jjdddfy)Nrr rr<r?r r) rrrrrr rgetrrr,)rrrrrrs rtest_axis_countzTestCount.test_axis_countZs|++dff1-MM'<rs} II D (1(%67\D8\D~#@ <*-(I[D[Dr