ixddlmZddlZddlZddlZddlZddlZddlZddlddl m Z dejk(Z ddZ d dZdZd Zd Zd Zd!d Zd Zd!dZGddej,ZGddej0j2ZGddeZGddeZGddeZGddej,Zej>e dGddej,Z e!dk(rejDyy)") OrderedDictN)*)Vector2PyPyctj|}t|d|dzdzD]M}tjd|ddz tjd|ddz }}|j ||fO|S)zwrandom_mask(size=(100,100)): return Mask Create a mask of the given size, with roughly half the bits set at random.r)pygameMaskrangerandomrandintset_at)sizemixys `/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/pygame/tests/mask_test.py random_maskrs~  DA 47T!W$) *~~aa1-v~~aa1/M1 !Q Hc"tj|j}|j}|rot |j D]Q}t |j D]3}|j|dz|dzf|k7s |j||fd5S|St |j D]N}t |j D]0}|j||fd|kDs|j||fd2P|S)Ng?r) r r get_size get_colorkeyr get_height get_widthget_atr)surface thresholdmaskkeyrrs rmaskFromSurfacer#s  ;;w'') *D    C w))+, +A7,,./ +>>1s7AG"45<KKA* + + K w))+, +A7,,./ +>>1a&)!,y8KKA* + + Krc|ddx}}|ddx}}|ddD]5\}}t||}t||}t||}t||}7tj||f||z dz||z dzfS)z.Creates a bounding rect from the given points.rrN)minmaxr Rect)pointsxminxmaxyminymaxrrs rcreate_bounding_rectr-*s)A,D4)A,D4qr 11d|1d|1d|1d|  ;;d|dTkAotd{Q%G HHrc2||f|dfd|fdffdDS)zCreates a generator which yields pairs of sizes. For each pair of sizes at least one of the sizes will have a 0 in it. rrrc3FK|]}D]}d|vsd|vs ||fywrN).0absizess r z"zero_size_pairs..?s- Eq5 EaAFa1fQF EF Es! !r2)widthheightr6s @rzero_size_pairsr:8s* V_uaj1f+v >E EE EErcX|j\}}d|dz df|dz |dz fd|dz ffS)zjReturns a tuple with the corner positions of the given mask. Clockwise from the top left corner. r/rr)r)r!r8r9s rcornersr<Bs? MMOME6 UQYNUQY $;a!_ MMrcv|jdz |jf|jdz |jdz f|j|jdz f|jdz |jdz f|j|jdz f|j|jf|j|jdz f|j|jf|jdz |jf|j|jf|jdz |jf|jdz |jdz ff S)zvReturns a tuple with the positions off of the corners of the given rect. Clockwise from the top left corner. r)lefttoprightbottom)rects r off_cornersrCKs Q! Q1 % DHHqL! aA& TXX\" TXX T[[1_% T[[! a% DKK Q $ Q a(  rc|3t|jt|j}no|j|j |j }t|j |jt|j|j}|jfd|DD]$}|j|j|||&|jy)zChecks to see if the given surface is filled with the given color. If an area_rect is provided, only check that area of the surface. Nc34K|]}D]}||f ywNr2r3rrx_ranges rr7z&assertSurfaceFilled..o!919AA99)r rr normalizeclipget_rectr>r@r?rAlock assertEqualrunlock)testcaserexpected_color area_recty_rangeposrHs @rassertSurfaceFilledrV`s ))+,**,-NN7#3#3#56   8 y'7'78 LLN99GW^^C0.#FG NNrcRt|jt|j}|j|j fd|DD]6}|j |r|j |j|||8|jy)zcChecks if the surface is filled with the given color. The ignore_rect area is not checked. c34K|]}D]}||f ywrFr2rGs rr7z0assertSurfaceFilledIgnoreArea..}rIrJN) r rrrKrN collidepointrOrrP)rQrrR ignore_rectrTrUrHs @rassertSurfaceFilledIgnoreArear[tsG%%'(GG&&()G LLN99K'',  !4nc JK NNrc|j}|j|j|j||j||j||j||j|d|y)z-Checks to see if the 2 given masks are equal.)msgr/N)countrOr overlap_area)rQm1m2r]m1_counts rassertMaskEqualrcsixxzH  3? 288:37 2??2v#>CHrc\ eZdZdZej eddZdZdZ ej eddZ dZ dZ d Z d Zd Zd Zd ZdZej eddZdZdZej eddZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"ej eddZ#ej edd Z$d!Z%ej edd"Z&ej edd#Z'd$Z(ej edd%Z)d&Z*d'Z+d(Z,ej edd)Z-d*Z.ej edd+Z/d,Z0ej edd-Z1d.Z2d/Z3d0Z4d1Z5ej edd2Z6d3Z7d4Z8ej edd5Z9d6Z:d7Z;ej edd8Z<ej edd9Z=d:Z>d;Z?d<Z@d=ZAd>ZBej edd?ZCd@ZDdAZEdBZFdCZGdDZHdEZIej eddFZJdGZKdHZLdIZMej eddJZNdKZOdLZPdMZQdNZRdOZSdPZTdQZUdRZVdSZWdTZXdUZYdVZZdWZ[dXZ\dYZ]dZZ^d[Z_d\Z`ej edd]Zadd^Zbdd_Zcdd`ZddaZedbZfdcZgddZhdeZidfZjdgZkdhZldiZmdjZndkZoej eddlZpej eddmZqdnZrdoZsdpZtdqZudrZvdsZwdtZxduZydvZzdwZ{ejdxZ}dyZ~dzZd{Zd|Zd}Zd~ZdZdZdZdZdZdZdZdZdZdZejdZejdZejdZejdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZej eddZdZejej eddZejdZejdZdZdZdZdZdZdZdZdZdZdZdZdZdZejej eddZejdZejej eddZejej edd„ZdÄZdĄZdńZdƄZdDŽZdȄZdɄZdʄZd˄Zd̄Zd̈́Zd΄ZdτZdЄZdфZd҄ZdӄZdԄZdՄZdքZdׄZd؄ZdلZdڄZdۄZd܄Zd݄ZdބZd߄ZdZdZy) MaskTypeTest) r/rrrrrrrrrjrjrjrjrrjrzSegfaults on pypyc2d}d}tjj|}tjj|}|j|tjj|j |j ||j |j ||j|tjj|j |j ||j |j |y)z:Ensure masks are created correctly without fill parameter.r rN)r r!r assertIsInstancerOr^r)selfexpected_count expected_sizemask1mask2s r test_maskzMaskTypeTest.test_masks    /   m 4 eV[[%5%56 7 )=9 eV[[%5%56 7 )=9rcdD]5}|jt5tj|}ddd7y#1swYBxYw)z=Ensure the mask constructor handles negative sizes correctly.)rirnrlN) assertRaises ValueErrorr r rurr!s rtest_mask__negative_sizez%MaskTypeTest.test_mask__negative_sizesG0 )D"":. ){{4( ) ) ) ) )s <A c|d\}}||f}||zdd}|jD]\}}d|}tjj||}|j |tjj||j |j |||j |j||y)z:Ensure masks are created correctly using the fill keyword.)%/rTFfill=fillNitemsr r!r rtrOr^r rur8r9rw fill_countsrrvr]r!s rtest_mask__fill_kwargz"MaskTypeTest.test_mask__fill_kwargs v "V^A6 $/$5$5$7 B D.$.C;;##M#=D  ! !$ (8(8# >   TZZ\>3 ?   T]]_mS A BrctddD]}tddD]}||z}||f}d|}tjj|d}|j |tjj||j |j |||j |j||y)aEnsures masks are created correctly using the fill keyword over a range of sizes. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) rBsize=TrN)r r r!r rtrOr^r)rur9r8rvrwr]r!s r$test_mask__fill_kwarg_bit_boundariesz1MaskTypeTest.test_mask__fill_kwarg_bit_boundariessAqk FFq" F!&!& m_-{{'' D'A%%dFKK,<,   TZZ\>3 ?   T]]_mS A Brcd\}}||f}||zdd}|jD]'\}}d|}tjj||}tjj||} |j |tjj||j | tjj||j |j |||j | j |||j |j|||j | j||*y)z:Ensure masks are created correctly using the size keyword.)ISrrrrr)rrNr) rur8r9rwrrrvr]rxrys rtest_mask__size_kwargz"MaskTypeTest.test_mask__size_kwargs v "V^A6 $/$5$5$7 C D.$.CKK$$$]$CEKK$$-d$CE  ! !%)9)93 ?  ! !%)9)93 ?   U[[]NC @   U[[]NC @   U^^-}c B   U^^-}c B CrcdD]}dD]}tjj||f}t|D]*}t|dz|dD]}|j ||f,|j t j |fD]K}|j |tjj|j||t|||My)z:Ensures copy works correctly with some bits set and unset. !?@Arr N) r r!r r rcopyrt assertIsNotrc)rur8r9r!rr mask_copys r test_copyzMaskTypeTest.test_copys. ;E2 ;{{''8u,A"1q5&!4, QF+,, #'))+tyy!?;I)))V[[5E5EF$$Y5#D)T:; ; ;rcLdD]}dD]}tjj||fd}|jtj|fD]K}|j |tjj|j ||t |||My)z0Ensures copy works correctly on a filled masked.rTrNr r!r rrtrrcrur8r9r!rs rtest_copy__fullzMaskTypeTest.test_copy__fulls. ;E2 ;{{''d'C#'))+tyy!?;I)))V[[5E5EF$$Y5#D)T:;  ; ;rcHdD]}dD]}tjj||f}|jtj|fD]K}|j |tjj|j ||t |||My)z.Ensures copy works correctly on an empty mask.rNrrs rtest_copy__emptyzMaskTypeTest.test_copy__emptys- ;E2 ;{{''8#'))+tyy!?;I)))V[[5E5EF$$Y5#D)T:;  ; ;rcd}d}tjjd}|jtj|f}|j ||D]}|j ||j |||j |j||j||j |j||j|y)z3Ensures copy makes an independent copy of the mask.)rr)rr rrN)r r!r rrrassertNotEqualr)ru mask_set_posmask_copy_set_posr! mask_copiesrs rtest_copy__independentz#MaskTypeTest.test_copy__independent's #{{(yy{DIIdO4  L!$ I   . /   Y -     . L0I      !23T[[AR5S  rcd}tjj|}|j|j |y)z,Ensure a mask's size is correctly retrieved.)]eN)r r!r rOr)rurwr!s r test_get_sizezMaskTypeTest.test_get_size<s1! {{ . -8rctjdd}dD]O}tjj|j|}|j }|j ||Qy)z!Ensures get_rect works correctly.r/rq rrN)r r'r!r rrMrO)ru expected_rectrr!rBs r test_get_rectzMaskTypeTest.test_get_rectCs] FH5 " 2D;;##M$6$6T#BD==?D   T= 1  2rc Vd}d}|d|dddgtddfi}d }tjj|}|j D]\\}}|D]R}|D]K} tj d |} t | || |jd i|| i} |j| | MT^y ) ziEnsures get_rect supports a single rect attribute kwarg. Tests all the rect attributes. ) rrr?r>rAr@centerxcenteryr8r9wh) topleft bottomlefttopright bottomrightmidtopmidleft midbottommidrightcenterr)rgffffff@)rg@gffffff@rrrrr/Nr2) rr r!r rr'setattrrMrO) ruRECT_SINGLE_VALUE_ATTRIBUTESRECT_DOUBLE_VALUE_ATTRIBUTESrect_attributesrr! attributesvalues attributevaluerrBs rtest_get_rect__one_kwargz%MaskTypeTest.test_get_rect__one_kwargOs ( $ ( $ )( (8c1Xwq!}*M  {{%"1"7"7"9 : J' : #:E$*KK$=MM9e<(4==>Iu+=>D$$T=9 : : :rc tjjd}tjdd}ddt ddd}|j D]\}}t ||||jd i|}|j||y) z9Ensures get_rect supports multiple rect attribute kwargs.rr/gffffff@rjr g @)rr?rNr2) r r!r r'rrrrMrO)rur!rkwargsattribrrBs rtest_get_rect__multiple_kwargsz+MaskTypeTest.test_get_rect__multiple_kwargss{{' FF3 2wq#?#\\^ 2MFE M65 1 2t}}&v& }-rc.tjjd}|jt5|j d}ddd|jt5|j d}dddy#1swY9xYw#1swYyxYw)z&Ensures get_rect only supports kwargs.rrrNrr r r!r r| TypeErrorrMrur!rBs rtest_get_rect__no_arg_supportz*MaskTypeTest.test_get_rect__no_arg_supports{{'   y ) $==#D $  y ) )==(D ) ) $ $ ) )sA?$B ?B Bctjjd}|jt5|j d}ddd|jt5|j d}ddd|jt5|j d}dddy#1swYjxYw#1swYFxYw#1swYyxYw) z(Ensures get_rect detects invalid kwargs.rrq)righteNrg)toplefrr )move)r r!r r|AttributeErrorrMrs r!test_get_rect__invalid_kwarg_namez.MaskTypeTest.test_get_rect__invalid_kwarg_names{{'   ~ . ,===+D ,  ~ . 0===/D 0  ~ . .==f=-D . .  , , 0 0 . .s#B1%B=C 1B:=C Cctjjd}|jt5|j d}ddd|jt5|j d}ddd|jt5|j d}ddd|jt5|j d }ddd|jt5|j d }ddd|jt5|j d }ddd|jt5|j d}dddy#1swY+xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxxYw#1swYyxYw)z/Ensures get_rect detects invalid kwarg formats.)rrq1)r@Nr)rArg)r)rr)r)rrr r)rr)rrrs r#test_get_rect__invalid_kwarg_formatz0MaskTypeTest.test_get_rect__invalid_kwarg_formats{{(   y ) ,==s=+D ,  y ) .===-D .  y ) 1===0D 1  y ) 3===2D 3  y ) 0==$=/D 0  y ) 7==I=6D 7  y ) .==1=-D . .% , , . . 1 1 3 3 0 0 7 7 . .sSE1%E>F F5F#%F/F;1E;>F FF #F,/F8;Gcd\}}tjj||f}tjj||fd}d}d}|dz |dz f}|j|j |||j|j |||j|j t |||j|j t ||y)z4Ensure individual mask bits are correctly retrieved.rrTrrr)rUN)r r!r rOrr)rur8r9mask0rxmask0_expected_bitmask1_expected_bitrUs r test_get_atzMaskTypeTest.test_get_ats v   %1   %t <qy&1*% c*,>? #.0BC gcl35GH '#,79KLrc0d\}}tjj||f}|jt5|j |dfddd|jt5|j d|fddd|jt5|j dddd|jt5|j ddddy#1swYxYw#1swYtxYw#1swYQxYw#1swYyxYw)zEnsure get_at() checks bounds.rqrrNrmrkr r!r r| IndexErrorrrur8r9r!s rtest_get_at__out_of_boundsz'MaskTypeTest.test_get_at__out_of_bounds v{{0   z * $ KK # $  z * % KKF $ %  z * ! KK  !  z * ! KK  ! ! $ $ % % ! ! ! !/C(-C4D D (C14C=D  Dc2d\}}tjj||f}tjj||fd}d}|j}d}|dz |dz f}|j |||j t |||j |j|||j |j||j |j|||j |j|y)z)Ensure individual mask bits are set to 1.rTrr)rUrN)r r!r r^rrrOr rur8r9rrxmask0_expected_countmask1_expected_count expected_bitrUs r test_set_atzMaskTypeTest.test_set_ats v   %1   %t < ${{} qy&1*% S,' \ : c*L9 (<= c*L9 (<=rc$d\}}tjj||f}tjj||fd}d}|jdz }d}|dz |dz f}|j |||j |||j |j |||j |j||j |j |||j |j|y)z)Ensure individual mask bits are set to 0.rqrTrrrNr r!r r^rrOrrs rtest_set_at__to_0zMaskTypeTest.test_set_at__to_0s v   %1   %t < ${{}q0 qy&1*% S,' S,' c*L9 (<= c*L9 (<=rcd\}}tjj||f}tjj||fd}d}|j}d}|dz |dz f}|j ||j ||j |j |||j |j||j |j |||j |j|y)zEnsure an offset overlap intersection is correctly calculated.rTrrroffset=rrotherrN) r r!r r^rORIGIN_OFFSETSr&rrO) rurxryrr mask1_size mask2_sizerr]rrs rtest_overlap__offsetz!MaskTypeTest.test_overlap__offsetOs-   t 4   t 4kkm kkm ^^% ^^% )) @FF8$Cq 1-s6!9a/@AL--eF-CK   [, <   U[[]K =   U[[]K =   U^^-z3 ?   U^^-z3 ? @rctjjdd}tjjdd}d}|j|d|j|d|j }|j }|j }|j }|j D]G}d|} |\} } t| d} d| k(rt| dzd} n!d| krt| dzd} n t| d} |j|t|}|j|| | f| |j|j || |j|j || |j|j || |j|j || |j|j|d| |j|j|d| Jy ) z_Ensure an offset overlap intersection is correctly calculated when (0, 0) bits not set.rTrrr/rr rN) r r!r rr^rr#r&rrrOr)rurxry unset_posrrr$r%rr]rr expected_y expected_xrs r$test_overlap__offset_with_unset_bitsz1MaskTypeTest.test_overlap__offset_with_unset_bitsfs   t 4   t 4  Y" Y"kkm kkm ^^% ^^% )) >FF8$CDAqQJAv Q] Q Q]  AY --wv?K   [:z*BC H   U[[]K =   U[[]K =   U^^-z3 ?   U^^-z3 ?   U\\)4a =   U\\)4a =+ >rctjjdd}|j}|j }d\}}||f}tjj|}|dz |dz f}|j |d} |j D]} d| } |j|| } |j| | |j|j|| |j|j| | |j|j || |j|j || |j|j|d| y)z_Ensure an offset overlap intersection is correctly calculated when there is no overlap.rTr)Crrr N) r r!r r^rrr#r assertIsNonerOr) rurxrr$mask2_wmask2_hr%ryset_posrrr]rs rtest_overlap__no_overlapz%MaskTypeTest.test_overlap__no_overlaps?   t 4kkm ^^%  w'    ,Q;! , W ))  (((>#11 GF!)IfX>C$'q 1$5s6!9a7H#IL"'--v">K$$[,D$$U[[]JD$$U[[]JD$$U^^%5y#F$$U^^%5y#F G G Grcd}d}tjj|}tj|}|j t 5|j ||}dddy#1swYyxYw)z8Ensure overlap handles invalid mask arguments correctly.rrr/N)r r!r Surfacer|rr)rurrr! invalid_maskrs rtest_overlap__invalid_mask_argz+MaskTypeTest.test_overlap__invalid_mask_argsc{{%~~d+   y ) =,,|VEnsures overlap_area handles offsets and boundaries correctly.rTrrrrr N)r r!r r^rr_rrO) rurxryrrr$r%rvr5rr]rHs r"test_overlap_area__offset_boundaryz/MaskTypeTest.test_overlap_area__offset_boundary9sO   t 4   d 3kkm kkm ^^% ^^% ]A   1 m^Q  A    @FF8$C!..ugfoFM   ]NC @   U[[]K =   U[[]K =   U^^-z3 ?   U^^-z3 ? @rc tddD]t}tddD]a}||f}||z}tjj|d}tjj|d}|j }|j }|j D]} d|d| } | |_|j|} | j| jz} |j|| } |j| | | |j|j|| |j|j|| |j|j|| |j|j|| dwy) zEnsures overlap_area handles masks of different sizes correctly. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) r rrTrrr8N)r r r!r rMr#rrLrrr_rOr^r)rur9r8r9r:rxryrKrLrr]rMexpected_overlap_countrHs r!test_overlap_area__bit_boundariesz.MaskTypeTest.test_overlap_area__bit_boundariesXslAqk GFq" G"FO "V^  (((> (((>((#11GF!)IfX>C$*EM#(::e#4L-9^^lnn-L*$)$6$6uf$EM$$]4JCP$$U[[]JD$$U[[]JD$$U^^%5y#F$$U^^%5y#FG G Grcd}d}tjj|}tj|}|j t 5|j ||}dddy#1swYyxYw)z=Ensure overlap_area handles invalid mask arguments correctly.rrr/N)r r!r r>r|rr_)rurrr!r?rHs r#test_overlap_area__invalid_mask_argz0MaskTypeTest.test_overlap_area__invalid_mask_arg|sg{{%~~d+   y ) D --lFCM D D DrAcd}d}tjj|}tjj|}|jt5|j ||}dddy#1swYyxYw)z?Ensure overlap_area handles invalid offset arguments correctly.)rr rCN)r r!r r|rr_)rurrrxryrHs r%test_overlap_area__invalid_offset_argz2MaskTypeTest.test_overlap_area__invalid_offset_argsi   &   &   y ) >!..uf=M > > >rEcNd}d}tjj|}dtjj|di}dD]Y}tjj||}|j}dD]}||f} d| } tjj||} | j} |j | |} | j ||}|j |tjj| t||| | |j| j| | |j|j|| |j| j|| |j|j|| "\y ) adEnsure overlap_mask's mask has correct bits set. Testing the different combinations of full/empty masks: (mask1-filled) 1 overlap_mask 1 (mask2-filled) (mask1-empty) 0 overlap_mask 1 (mask2-filled) (mask1-filled) 1 overlap_mask 0 (mask2-empty) (mask1-empty) 0 overlap_mask 0 (mask2-empty) rr/rTrrrr!N) r r!r r^r overlap_maskrtrcrOr)rurwrrexpected_masksrryrrr"r]rxr expected_maskrZs rtest_overlap_maskzMaskTypeTest.test_overlap_masksu !;;++M:& (8(8T(8(RS" GEKK$$]$?E++-K& GenSEl ((U(C#kkm . 2 238H I $11f1M %%lFKK4D4DcJlM3G   SA   SA  !1=#F  !1=#F! G  Grc Ztjjdd}tjjdd}|j}|j}|j }|j }|j |d}t dD]=}t dD]-} |j|j|| fdd |d | d /?t dD]>}t d dD]-} |j|j|| fd d |d | d /@|j|j||j|j||j|j ||j|j |y)z0Ensure overlap_mask's mask has correct bits set.2r`Tr)i, rmr`rar(, )rqrN) r r!r r^rrZr rOr) rurxryrrr$r%mask3rjs rtest_overlap_mask__bits_setz(MaskTypeTest.test_overlap_mask__bits_sets    5    6kkm kkm ^^% ^^% ""5'2r IA2Y I  q!f!5qAaS1#Q-H I Ir IA2r] I  q!f!5qAaS1#Q-H I I  4  4 ):6 ):6rctjjdd}tjjdd}|j}|j}|j }|j }tj|}|j }|j } |j D]:} d| } | | _|j| } |j|jtj| jd| j|j|| } |j| tjj| t|| || |j|j|| |j|j|| |j|j || |j|j || =y)z=Ensure an offset overlap_mask's mask is correctly calculated.rTrrr N)r r!r r^rrMr#rrLcleardrawrrZrtrcrO)rurxryrrr$r%r\rKrLrr]rMrZs rtest_overlap_mask__offsetz&MaskTypeTest.test_overlap_mask__offsets   t 4   t 4kkm kkm ^^% ^^%  J/   )) @FF8$C"EM ::e,L    !    L--D9<;O;O !--eVEnsures overlap_mask handles offsets and boundaries correctly.) rTr)rqrrrr N)r r!r r^rrZrtrO) rurxryrrr$r%rvrwr5rr]rZs r"test_overlap_mask__offset_boundaryz/MaskTypeTest.test_overlap_mask__offset_boundarys   d 3   t 4kkm kkm ^^% ^^% " ]A   1 m^Q  A    @FF8$C --eV3 G   \224mS I   U[[]K =   U[[]K =   U^^-z3 ?   U^^-z3 ? @rc tddD]}tddD]}||f}||z}tjj|d}tjj|d}tj|}|j }|j } |j D]=} d|d| } | | _|j| } |j|jtj| jd| j |j|| } |j| tjj| t|| || |j|j|| |j|j|| |j|j!|| |j|j!|| @y) zEnsures overlap_mask handles masks of different sizes correctly. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) r rrTrrr8N)r r r!r rMr#rrLrirjrrZrtrcrOr^r)rur9r8r9r:rxryr\rKrLrr]rMrZs r!test_overlap_mask__bit_boundariesz.MaskTypeTest.test_overlap_mask__bit_boundaries6sAqk GFq" G"FO "V^  (((> (((> & I 6 ((#11GF!)IfX>C$*EM#(::e#4L!'')!&& L$5$5DAr|rrZ)rurrr!r?rZs r#test_overlap_mask__invalid_mask_argz0MaskTypeTest.test_overlap_mask__invalid_mask_arg_sg{{%~~d+   y ) C,,\6BL C C CrAcd}d}tjj|}tjj|}|jt5|j ||}dddy#1swYyxYw)z?Ensure overlap_mask handles invalid offset arguments correctly.rr rCN)r r!r r|rrZ)rurrrxryrZs r%test_overlap_mask__invalid_offset_argz2MaskTypeTest.test_overlap_mask__invalid_offset_argisi   &   &   y ) = --eVz/MaskTypeTest.test_mask_access..sahhw.?rc(jddS)Nrmrrrsrrz/MaskTypeTest.test_mask_access..ahhw.Brc(jddS)Nrarrrrsrrz/MaskTypeTest.test_mask_access..rrc(jddS)Nrrarrrsrrz/MaskTypeTest.test_mask_access..rrN)r r rrOrr|rrurs @rtest_mask_accesszMaskTypeTest.test_mask_accessss KK !  &)1-  &)1- *&?@ *&BC *&BC *&BCrcd\}}||z}||f}tjj|}|j|j |j ||j |j |y)zEnsure a mask can be filled.rpN)r r!r rrOr^r)rur8r9rvrwr!s r test_fillzMaskTypeTest.test_fillsf v {{ .  ~6 -8rc tddD]q}tddD]`}tjj||f}||z}|j |j |j |d|d|dbsy)zEnsures masks of different sizes are filled correctly. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) rrrsize=(rcrdN)r r r!r rrOr^)rur9r8r!rvs rtest_fill__bit_boundariesz&MaskTypeTest.test_fill__bit_boundariessAqk Fq" {{''8!&   JJL.F5'F812M   rcd}d}tjj|d}|j|j |j ||j |j |y)zEnsure a mask can be cleared.rrTrN)r r!r rirOr^rrurvrwr!s r test_clearzMaskTypeTest.test_clearsZ {{ D9  ~6 -8rc d}tddD]n}tddD]]}tjj||fd}|j |j |j |d|d|d _py ) zEnsures masks of different sizes are cleared correctly. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) rrrrTrrrcrdN)r r r!r rirOr^)rurvr9r8r!s rtest_clear__bit_boundariesz'MaskTypeTest.test_clear__bit_boundariessAqk Fq" {{''d'C   JJL.F5'F812M   rcd}||f}tjj|}tjj|d}||z}d}t|D]3}|dz}|dz }||f}|j ||j |d5|j |j |j |j||j |j||j |j||j |j|t|D]O}||f}d|} |j |j|d| |j |j|d| Qy)zEnsure a mask can be inverted.rTrrrzpos=N) r r!r r rinvertrOr^rr) rusiderwrxryexpected_count1expected_count2rrUr]s r test_invertzMaskTypeTest.test_invertsbt    /   T :+t !A q O q Oa&C LL  LLa  !    8 8 )=9 )=9t 8Aa&C,C   U\\#.3 7   U\\#.3 7  8rcd}d}tjj|d}|j|j |j ||j |j |y)z#Ensure a full mask can be inverted.r+aTrNr r!r rrOr^rrs rtest_invert__fullzMaskTypeTest.test_invert__fullsZ {{ D9  ~6 -8rcd\}}||f}||z}tjj|}|j|j |j ||j |j |y)z%Ensure an empty mask can be inverted.rNr)rur8r9rwrvr!s rtest_invert__emptyzMaskTypeTest.test_invert__emptysf v {{ .  ~6 -8rc$dD]}tddD]z}tddD]i}tjj||f|}|rdn||z}|j |j |j |d|d|d |d k|y ) zEnsures masks of different sizes are inverted correctly. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) rrrrrrr, size=(rcrdN)r r r!r rrOr^)rurr9r8r!rvs rtest_invert__bit_boundariesz(MaskTypeTest.test_invert__bit_boundariess" D1+ "1b\ E!;;++UFO$+GD*.QEFNNKKM$$ &vXeWBvha@   rc|d\}}||f}dD]-}tjj||}|r||znd}t|dz |dzD]}t|dz |dzD]}||f} |r||znd} d| } |j | } |j | tjj| |j | j| | |j | j| |j |j|| |j |j|| 0y) zEnsure a mask can be scaled.)r=rrrrar)scaleN) r r!r r rrtrOr^r) rur8r9 original_sizer original_maskoriginal_countnew_wnew_hrwrvr]r!s r test_scalezMaskTypeTest.test_scalesF v ! SD"KK,,],FM/3UV^Nurz52:6 S"6B; < SE%*ENM6:UU]N!-1C(..].CD))$ 0@0@#F$$TZZ\>3G$$T]]_mD$$]%8%8%:NCP$$]%;%;%=}cR S S  Srctjd}|jt5|j dddd|jt5|j t ddddd|jt5|j ddddy#1swYrxYw#1swYExYw#1swYyxYw)z.Ensure scale handles negative sizes correctly.drrlNrjra)rarj)r r r|r}rrrur!s rtest_scale__negative_sizez&MaskTypeTest.test_scale__negative_size,s{{:&   z * ! JJx  !  z * ( JJwr2 ' (  z * ! JJx  ! !  ! ! ( ( ! !s#B.B:C.B7:CCcJd}d}tjj|d}dtjj|i}dD]}tjj||}|j}dD]}||f} d| } tjj||} |j | |} | j ||t || | | |j|j|| |j|j|| y) aAEnsure a mask can be drawn onto another mask. Testing the different combinations of full/empty masks: (mask1-filled) 1 draw 1 (mask2-filled) (mask1-empty) 0 draw 1 (mask2-filled) (mask1-filled) 1 draw 0 (mask2-empty) (mask1-empty) 0 draw 0 (mask2-empty) rr/Tr)FFrrN) r r!r r^rrjrcrOr rurwrrr[rryrrr"r]rxr\s r test_drawzMaskTypeTest.test_draw9s !;;++M+E(&++*:*:=*IJ" GEKK$$]$?E++-K& GenSEl ((U(C . 2 238H I  5&)e]C@   SA  !1=#F G  Grcjtjjd}tjjdd}|j}|j }tj|j }|j }|j }|j D]}d|} ||_|j|} |jt| j| jD]:} t| j| jD]} |j| | f<|j|j!||t#|||| |j%|j|| |j%|j || y)z5Ensure an offset mask can be drawn onto another mask.rrTrr r!N)r r!r r^rrMr#rrLrir r>r@r?rArrjrcrO rurxryrr%r\rKrLrr]rMrrs rtest_draw__offsetzMaskTypeTest.test_draw__offsetYsm   )   t 4kkm ^^%  ENN$45   )) @FF8$C"EM ::e,L    !<,,l.@.@A 1|//1D1DE1A!((!Q01 1 KKM JJU6J 2 D% <   U[[]K =   U^^-z3 ?' @rc tjjd}tjjdd}tj|j}|j }|j }|j dd}dD]}t ||t|||j}d|} |j|} |jt| j| jD]:} t| j| jD]} |j!| | f<|j|j#||t%|||| y) aEnsure an offset mask can be drawn onto another mask. Testing the specific case of: -both masks are wider than 32 bits -a positive offset is used -the mask calling draw() is wider than the mask passed in rmrnTrrorpr N)r r!r rrMrqrrrrrLrir r>r@r?rArrjrc rurxryr\rKrLrsrtrr]rMrrs rtest_draw__specific_offsetsz(MaskTypeTest.test_draw__specific_offsetszsB   )   t 4 ENN$45   mmB+ J =F E67;#? @]]FF8$C ::e,L    !<,,l.@.@A 1|//1D1DE1A!((!Q01 1 KKM JJuf % D% r@r?rArrjrcrOr^rrur9r8r9r:rxryr\rKrLrr]rMrrs rtest_draw__bit_boundariesz&MaskTypeTest.test_draw__bit_boundariessAqk! GFq" G"FO "V^  ((3 (((> & I 6 ((#11GF!)IfX>C$*EM#(::e#4L!'') #<#4#4l6H6HI9!&|'7'79L9L!M9A)00!Q899KKMJJuf-#D%D$$U[[]JD$$U^^%5y#F)G G! Grcd}d}tjj|}tj|}|j t 5|j ||dddy#1swYyxYw)z5Ensure draw handles invalid mask arguments correctly.rr/N)r r!r r>r|rrjrurrr!r?s rtest_draw__invalid_mask_argz(MaskTypeTest.test_draw__invalid_mask_args_{{%~~d+   y ) , IIlF + , , ,rAcd}d}tjj|}tjj|}|jt5|j ||dddy#1swYyxYw)z7Ensure draw handles invalid offset arguments correctly.rrCN)r r!r r|rrjrurrrxrys rtest_draw__invalid_offset_argz*MaskTypeTest.test_draw__invalid_offset_argse   &   &   y ) & JJuf % & & &rEcJd}d}tjj|}dtjj|di}dD]}tjj||}|j}dD]}||f} d| } tjj||} |j | |} | j ||t || | | |j|j|| |j|j|| y)a=Ensure a mask can erase another mask. Testing the different combinations of full/empty masks: (mask1-filled) 1 erase 1 (mask2-filled) (mask1-empty) 0 erase 1 (mask2-filled) (mask1-filled) 1 erase 0 (mask2-empty) (mask1-empty) 0 erase 0 (mask2-empty) rr/rTrrN) r r!r r^rerasercrOrrs r test_erasezMaskTypeTest.test_erases !;;++M:')9)9-d)9)ST" GEKK$$]$?E++-K& GenSEl ((U(C . 2 238H I  E6*e]C@   SA  !1=#F G  Grcltjjd}tjjdd}|j}|j }tj|j }|j }|j }|j D]}d|} ||_|j|} |jt| j| jD];} t| j| jD]} |j| | fd=|j|j!||t#|||| |j%|j|| |j%|j || y) z-Ensure an offset mask can erase another mask.rrTrr rr!N)r r!r r^rrMr#rrLrr r>r@r?rArrrcrOrs rtest_erase__offsetzMaskTypeTest.test_erase__offsetso   )   t 4kkm ^^%  ENN$45   )) @FF8$C"EM ::e,L    <,,l.@.@A 4|//1D1DE4A!((!Q34 4 JJL KKeFK 3 D% <   U[[]K =   U^^-z3 ?' @rc tjjd}tjjdd}tj|j}|j }|j }|j dd}dD]}t ||t|||j}d|} |j|} |jt| j| jD];} t| j| jD]} |j!| | fd=|j|j#|t%|t'|||| y ) zEnsure an offset mask can erase another mask. Testing the specific case of: -both masks are wider than 32 bits -a positive offset is used -the mask calling erase() is wider than the mask passed in rmrnTrrorpr rN)r r!r rrMrqrrrrrLrr r>r@r?rArrrrcrs rtest_erase__specific_offsetsz)MaskTypeTest.test_erase__specific_offsets<sH   )   t 4 ENN$45   mmB+ J =F E67;#? @]]FF8$C ::e,L    <,,l.@.@A 4|//1D1DE4A!((!Q34 4 JJL KKwv / D% r@r?rArrrcrOr^rrs rtest_erase__bit_boundariesz'MaskTypeTest.test_erase__bit_boundaries~sAqk! GFq" G"FO "V^  ((3 (((> & I 6 ((#11GF!)IfX>C$*EM#(::e#4L!&&( #<#4#4l6H6HI<!&|'7'79L9L!Mr|rrrs rtest_erase__invalid_mask_argz)MaskTypeTest.test_erase__invalid_mask_args_{{%~~d+   y ) - JJ|V , - - -rAcd}d}tjj|}tjj|}|jt5|j ||dddy#1swYyxYw)z8Ensure erase handles invalid offset arguments correctly.r4rCN)r r!r r|rrrs rtest_erase__invalid_offset_argz+MaskTypeTest.test_erase__invalid_offset_argse   &   &   y ) ' KKv & ' ' 'rEc&d}||f}d}tjj|}t|D]}|dz }|j ||f|j }|j |||j |j|y)z/Ensure a mask's set bits are correctly counted.r-rrN)r r!r r rr^rOr)rurrwrvr!rr^s r test_countzMaskTypeTest.test_countst {{ .t A a N KKA    / -8rcdD]}tddD]}tddD]tjj|f|}|r|znd}fdt|DD]b}|r|j |d|dz}n|j |d|dz }|j }|j ||d|d d |d |dy ) zEnsures the set bits of different sized masks are counted correctly. Tests masks of different sizes, including: -masks 31 to 33 bits wide (32 bit boundaries) -masks 63 to 65 bits wide (64 bit boundaries) rrrrrrc3FK|]}tD]}||f ywrFr r3rrr8s rr7z:MaskTypeTest.test_count__bit_boundaries..s%T1uU|T!ATT!rrrcz), pos=N)r r r!r rr^rO)rurr9r!rvrUr^r8s @rtest_count__bit_boundariesz'MaskTypeTest.test_count__bit_boundariess" D1+ "1b\E!;;++UFO$+GD7;UV^N Uf T KKQ/*a/N KKQ/*a/N $ ((!*#D6%6('#O   rcd\}}||f}||z}tjj|d}|j}|j |||j |j |y)z4Ensure a full mask's set bits are correctly counted.)rrTrNr r!r r^rOr)rur8r9rwrvr!r^s rtest_count__full_maskz"MaskTypeTest.test_count__full_masksg v {{ D9  / -8rcd}d}tjj|}|j}|j |||j |j |y)z6Ensure an empty mask's set bits are correctly counted.rrNr)rurvrwr!r^s rtest_count__empty_maskz#MaskTypeTest.test_count__empty_masksQ {{ .  / -8rctjjdd}|jj}|j }|j ||y)z8Ensure a filled mask's centroid is correctly calculated.rTrN)r r!r rMrcentroidrO)rur!expected_centroidrs r test_centroidzMaskTypeTest.test_centroidsI{{T2 MMO22==? #45rcd}d}tjj|}|j}|j |||j |j |y)z8Ensure an empty mask's centroid is correctly calculated.r/)rgN)r r!r rrOr)rurrwr!rs rtest_centroid__empty_maskz&MaskTypeTest.test_centroid__empty_mask sS"" {{ .==? #45 -8rc(d\}}tjj||f}t|D]^}|j t|D]>}|j ||f|dz|f}|j }|j||@`y)z`Ensure a mask's centroid is correctly calculated when setting points along a single row.rr Nr r!r r rirrrOrur8r9r!rrrrs rtest_centroid__single_rowz&MaskTypeTest.test_centroid__single_rows v{{0v >A JJL5\ > QF#%&!VQK!==?  +<=  > >rcVd\}}tjj||f}td|D]t}|j t|D]T}|j |df|j ||f|dz|dzf}|j }|j||Vvy)z\Ensure a mask's centroid is correctly calculated when setting points along two rows.rrrr Nrrs rtest_centroid__two_rowsz$MaskTypeTest.test_centroid__two_rows(s v{{0q&! >A JJL5\ > QF# QF#%&!VQ!V$4!==?  +<= > >rc(d\}}tjj||f}t|D]^}|j t|D]>}|j ||f||dzf}|j }|j||@`y)zcEnsure a mask's centroid is correctly calculated when setting points along a single column.rr Nrrur8r9r!rrrrs rtest_centroid__single_columnz)MaskTypeTest.test_centroid__single_column;s v{{0u >A JJL6] > QF#%&QK!==?  +<=  > >rcVd\}}tjj||f}td|D]t}|j t|D]T}|j d|f|j ||f|dz|dzf}|j }|j||Vvy)z_Ensure a mask's centroid is correctly calculated when setting points along two columns.rrrr Nrrs rtest_centroid__two_columnsz'MaskTypeTest.test_centroid__two_columnsLs v{{0q% >A JJL6] > QF# QF#%&!VQ!V$4!==?  +<= > >rctjjd}|jj}t |D]}|j ||j}|j||y)zREnsure a mask's centroid is correctly calculated when its corners are set.rN) r r!r rMrr<rrrO)rur!rrtrs rtest_centroid__all_cornersz'MaskTypeTest.test_centroid__all_corners_sg{{' MMO22dm F KK  ==? #45rctjjd}|j}t |}t |D]\}}||dzdD]}|j |j||j||d|dk(r|dt|d|dz dzf}n4|d|dk(rt|d|dz dz|df}n |j}|j}|j||y)zWEnsure a mask's centroid is correctly calculated when only two corners are set.rrNrr ) r r!r rMr< enumeraterirabsrrrO) rur! mask_rect mask_cornersrcorner1corner2rrs rtest_centroid__two_cornersz'MaskTypeTest.test_centroid__two_cornersls{{'MMO t} #L1 >JAw'A0 >  G$ G$1:+)0Sgaj9P5QUV5V(W%QZ71:-),WQZ'!*-D)E)JGTUJ(W%(1(8(8%==?  +<= > >rcd}d}tj|}tjj|}|j }|j |t |j||y)z:Ensure a mask's orientation angle is correctly calculated.gFrN)r r>r! from_surfaceanglertfloatrO)ruexpected_anglerwrr!r s r test_anglezMaskTypeTest.test_anglesY" ../{{''0  eU+ /rcd}d}tjj|}|j}|j |t |j |||j|j|y)z5Ensure an empty mask's angle is correctly calculated.)krN) r r!r r rtr assertAlmostEqualrOr)rur rwr!r s rtest_angle__empty_maskz#MaskTypeTest.test_angle__empty_masksa! {{ .  eU+ un5 -8rctjd}|j|jd|j |j|jdtjdd}|j |d|j|jd|j |j|jd |j|d |j|jd |j|j|jdy ) z%Test fill, clear, invert, draw, eraserri'rTrr_i&rr/N) r r rOr^rrrrjri)rurras r test_drawingzMaskTypeTest.test_drawings KK # A&  E* [[ - H D)   C( r6 C(   A&rctjd}|j|jg|j dd|j|jdg|j dd|j|jddg|j dd|j|jgd|j|jd gd y )  )rrrra rarqrq)rrrrrr )every)rrrN)r r rOoutlinerrs r test_outlinezMaskTypeTest.test_outlines KK ! b) 1 xj1 1 2 3 1  IIKK  +-KLrc gd}|D]}tj|}|D]}tj|}|j|}|j|tjjdD]M}|j |j ||j ||j |zdz Oy)N)rg)rr)rrrrfr)r r convolvertr!rOr)rur6s1r`s2raors rtest_convolve__sizez MaskTypeTest.test_convolve__sizes8 BRB [[_KKO%%a)9)9:A$$ Qq)9BKKM!a/ 1::a=1  mV[[-=-=>a/rc~td}tjd}|jdtjd}tjd}|j |||j |d|j |tjjt||||j|j|j ||tdd|j |d|j |tjjt|||y ) zGchecks that convolution modifies only the correct portion of the outputrr r r/r_rgra)r"outputrrN) rr r rr rjrtr!rcrir)rurr&r#tests rtest_convolve__with_outputz'MaskTypeTest.test_convolve__with_outputs  ! KK   KK !{{8$ 1a !V a!1!12a&    1WR_ = !X a!1!12a&rctjdd}d}|D]b\}}|j|d|}|j|tjj|j |j |dy)Nr*Tr))rrr)rr r))roror))r1r)r r r rtr!rOr^)rufullpts_dataptrvr's rtest_convolve__out_of_rangez(MaskTypeTest.test_convolve__out_of_rangesr{{6-K"* D B MM$b9M  ! !-1A1A B   ]002N C  Drc td}td}|j|}|j|tjj t |jdD]c}t |jdD]B}|j|j||fdk(|j||dz |dz fduDey)z#Tests the definition of convolutionrrrcN) rr rtr r!r r rrOrr)rur`raconvrrfs r test_convolvezMaskTypeTest.test_convolves $  ${{2 dFKK$4$45t}}q)* A4==?1-.   KKA'1,bjja"fa"f=M.NRV.V  rctjj||fd}|j|dz|dzfd|r$|j |||j |S|j |||S)NTrr r)r r!r rrrrj)rur!rrUinversepatterns r_draw_component_pattern_boxz(MaskTypeTest._draw_component_pattern_boxss ++""D$.ks<41aQWWY 1 bit. )qr-r)rqr rTrrrrN) r r!r r rr^rjrLrtrOrr_)rurwrp_widthp_heightr1rWrXrOrNrrrvr!s r(test_connected_component__multi_set_bitsz5MaskTypeTest.test_connected_component__multi_set_bitss"  ((7 "%%*$!8TAX.!;;++Wh,?d+Kq(# 7A1g& 7AE7Q;//$++QFA6 7 7 *//1+_=009 dFKK$4$45 ~6 -8    . @. ,,.? //1=A   & &'7 I  rcld\}}||f}tjj|d}|dz|dzf}|j|d|j }d}|j |}|j |tjj|j|j ||j|j||j|j ||j|j||j|j|dy)zpEnsure a mask's connected component is correctly calculated when the coordinate's bit is unset. )mrTrr rN) r r!r rr^rLrtrOrr) rur8r9rwrr(rrvr!s r#test_connected_component__unset_bitz0MaskTypeTest.test_connected_component__unset_bits ! v  ((T(B aZ1- Y*&,,.00; dFKK$4$45 ~6 -8 ,,.? //1=A --i8!$88= >   ]002N C   ]335} E  F > >s #CC ctjd}|j|jg|j }|j |j |j |jdd|jdd|j }|j}|jd}|jd}|jd}|j |j |dj |j |dj d|j |dj d|j|g|jdd|j }|j d}|j d }|j}|jd}|jd }|jd}|j |j d|j |j d|j |j d|j t|d|j t|d|j t|d|j t|d|D],} |j| tjj.y ) rrr/rrgr rr)rwrwr*)minimumN) r r assertListEqualconnected_componentsrLrOr^rlenrtr!) rurcompcompscomps1comps2comps3comp1comp2r!s rtest_connected_componentsz&MaskTypeTest.test_connected_components sZ KK ! Q335r:$$& DJJL1  $$&&&(''*''*''* uQx~~'78 *A. *A. VR( $$&%%f-%%f-&&(''*'''2''* q) * * UQ' Va( Va( Va( :D  ! !$ (8(8 9 :rcg}d}d}tjj|}|jd}|j |||j |j ||j |j|y)zEnsures connected_components() properly handles negative min values when the mask is empty. Negative and zero values for the min parameter (minimum number of bits per connected component) equate to setting it to one. r)rrrjN)r r!r rhrgrOr^r)ruexpected_compsr:r9r!connected_compss r7test_connected_components__negative_min_with_empty_maskzDMaskTypeTest.test_connected_components__negative_min_with_empty_mask> su  {{ *33B7 _n= z2 )4rcfd}tjj|d}|j}d}|j d}|j t ||t||d||j |j||j |j|y)zEnsures connected_components() properly handles negative min values when the mask is full. Negative and zero values for the min parameter (minimum number of bits per connected component) equate to setting it to one. )rrqTrrrorN) r r!r r^rhrOrircr)rur9r!r: expected_lenrts r6test_connected_components__negative_min_with_full_maskzCMaskTypeTest.test_connected_components__negative_min_with_full_maskR s {{ 5ZZ\  33B7 _-|<oa0$7 z2 )4rctd}tjj|}i}t|D]H}|j |tjj|}|j ||||<J|ddz|ddzf}|j |tjj|}|j ||||<|j }|j d}|jt|t||D]_} d} t|jD]*} | j| sd} t|| || || =n|j| d a|j|j ||j|j|y ) aEnsures connected_components() properly handles negative min values when the mask has some bits set. Negative and zero values for the min parameter (minimum number of bits per connected component) equate to setting it to one. )rrrr rr1FTzmissing component for pt=N)r r!r r<rr^rhrOrituplekeysrrc assertTruer) rur9r!rsrtnew_maskrr:rtrjfoundr4s r:test_connected_components__negative_min_with_some_bits_setzGMaskTypeTest.test_connected_components__negative_min_with_some_bits_setg s {{ *dm .F KK {{'' 2H OOF #%-N6 "  .A,!#Yq\Q%67 F;;##I.!)vZZ\ 33B7 _-s>/BC# EDEN//12 ;;r? E#D$r0BC&r*   OOE%>rd#C D E z2 )4rc0g}|jd|jd|jd|jd|jd|jd|D]\}}g}tj|}|D]4}|jt||D]}|j |6t |t }|j}|jt |jt |d|y ) z+Ensures get_bounding_rects works correctly.)r))r/rhrf)r/))rr)rr ))r/rhrfrgr rrr)r=)))r rrgrrr*)r=))rrrr)r~))rrrrfrgrr)r=))r/)r)r)r0)r)r"rN) rJr r r-rsortedrzget_bounding_rectsrg) ru mask_datarrect_point_tuplesrectsr! rect_pointsr4expected_rectss rtest_get_bounding_rectsz$MaskTypeTest.test_get_bounding_rects s5                    (1  #D#E;;t$D0 $  1+>?%$BKKO$ $ $Eu5N++-E  t..0e<v  rc6d}d}tj|td}tjd}tjddftjddff}|D]\}}|j |tj j || }|j|} |j| |ts%|jtj| ||j| j|t|| |y ) z8Ensures empty and full masks can be drawn onto surfaces.rrrrredwhiteTblackFrN)r r>SRCALPHAColorrr!r to_surfaceassertIsIS_PYPYrOsys getrefcountrrV) ruexpected_ref_countrr surface_color test_fillsrRrr!rs rtest_to_surfacezMaskTypeTest.test_to_surface s..x4 U+ ||G,d3fll76KU5ST $. B ND LL ';;##Dt#4D1J MM*g .  !<>PQ   Z002D 9 j. A Brcd}t}d}d}tjddftjddff}|D] \}}tjj ||}d D]} | r|j d } n|j } |j | tjts%|jtj| ||j| j|z|j| j||j| j|t!|| |y ) zAEnsures empty and full masks can be drawn onto a created surface.r rrrTrFrrNrr rr!r rrtr>rrOrrr| get_flags get_bitsizerrV) rur expected_flagexpected_depthrrrRrr!use_argrs rtest_to_surface__create_surfacez,MaskTypeTest.test_to_surface__create_surface( s# ||G,d3fll76KU5ST $. F ND;;##Dt#4D( F!%!6J!%!2J%%j&..A$$S__Z%@BTU 4 4 6 FG  !7!7!9>J  !4!4!6=#D*nE F Frc*d}tjd}tjd}d}tjj|d}tj|}d|i}dD]}|j ||r|j d i|} n|j |d} |j| |ts%|jtj| ||j| j|t|| |y ) z/Ensures to_surface accepts a surface arg/kwarg.rrrr=TrrrNr2)r rr!r r>rrrrrOrrrrV) rurrRrrr!rr use_kwargsrs rtest_to_surface__surface_paramz+MaskTypeTest.test_to_surface__surface_paramA sg. U+ {{40..&W%' BJ LL ',T__6v6 !__VI->? MM*g .  !<>PQ   Z002D 9 j. A Brcd}t}d}tjd}d}tjj |d}tj |||}|j |d|i}dD]} | r|jd i|} n|jd |d} |j| tj ts%|jtj| ||j| j|z|j| j||j| j!|t#|| |y ) z2Ensures to_surface accepts a setsurface arg/kwarg.r rrr=Tr setsurfacerNr2rr rr!r r>rrrtrrOrrr|rrrrV) rurrrrRrr!rrrrs r!test_to_surface__setsurface_paramz.MaskTypeTest.test_to_surface__setsurface_paramY s+ e,{{40^^D-H ' +' BJ,T__6v6 !__T6,3GH  ! !*fnn =  !<>PQ OOJ002]B C   Z335~ F   Z002D 9 j. A Brcd}t}d}tjd}d}tjj |}tj |||}|j |d|i}dD]} | r|jdi|} n|jdd|d} |j| tj ts%|jtj| ||j| j|z|j| j||j| j!|t#|| |y) z4Ensures to_surface accepts a unsetsurface arg/kwarg.r rrr= unsetsurfacerNr2r) rurrrrRrr!rrrrs r#test_to_surface__unsetsurface_paramz0MaskTypeTest.test_to_surface__unsetsurface_paramt s* e,{{%~~dM>J .) ,/' BJ,T__6v6 !__T49OP  ! !*fnn =  !<>PQ OOJ002]B C   Z335~ F   Z002D 9 j. A Brcjd}t}d}tjd}d}tjj |d}d|i}dD]}|r|j d i|} n|j d d d |d} |j | tjts%|jtj| ||j| j|z|j| j||j| j|t!|| |y ) z0Ensures to_surface accepts a setcolor arg/kwarg.r rrr=TrsetcolorrNr2r rurrrrRrr!rrrs rtest_to_surface__setcolor_paramz,MaskTypeTest.test_to_surface__setcolor_param s e,{{40n-' BJ,T__6v6 !__T4vj?QR  ! !*fnn =  !<>PQ OOJ002]B C   Z335~ F   Z002D 9 j. A Brctjd}d}tjj|d}|j dddd}|j |j |t|||y)z(Ensures the default setcolor is correct.rrTrN)rrr unsetcolorr rr!r rrOrrVrurRrr!rs r!test_to_surface__setcolor_defaultz.MaskTypeTest.test_to_surface__setcolor_default sqg.{{40__T%  ,,.5D*n=rc hd}t}d}tjd}d}tjj |}d|i}dD]}|r|j di|} n|j dddd|d} |j | tjts%|jtj| ||j| j|z|j| j||j| j|t!|| |y) z2Ensures to_surface accepts a unsetcolor arg/kwarg.r rrr=rrNr2rrs r!test_to_surface__unsetcolor_paramz.MaskTypeTest.test_to_surface__unsetcolor_param s e,{{%/' BJ,T__6v6 !__$dF<,@   ! !*fnn =  !<>PQ OOJ002]B C   Z335~ F   Z002D 9 j. A Brctjd}d}tjj|}|j dddd}|j |j |t|||y)z*Ensures the default unsetcolor is correct.rrN)rrrrrrs r#test_to_surface__unsetcolor_defaultz0MaskTypeTest.test_to_surface__unsetcolor_default slg.{{%__Tt%  ,,.5D*n=rc zd}t}d}d}tjd}d}d}tjj |}d|i} dD]} | r|} |j d i| } n|} |j d d d d d | d} |j | tjts%|jtj| ||j| j|z|j| j||j| j|t!|| | y ) z,Ensures to_surface accepts a dest arg/kwarg.r rrrrrrr/r=destrNr2r) rurrrdefault_surface_colordefault_unsetcolorrrr!rrrRrs rtest_to_surface__dest_paramz(MaskTypeTest.test_to_surface__dest_param s)  ,#\\'2{{%$' BJ!3,T__6v6 !6!__$dD&.   ! !*fnn =  !<>PQ OOJ002]B C   Z335~ F   Z002D 9 j. A% Brctjd}tjd}d}tjj|d}|j }|ddz|ddzf}tj |t d }|j||j|d d d }|j|||j|j|t||||t||||y ) z$Ensures the default dest is correct.rrrTrrr rrNrrrr rr!r rMr>rrrrrOrrVr[ rurRrr9r!r surf_sizerrs rtest_to_surface__dest_defaultz*MaskTypeTest.test_to_surface__dest_default sg. U+  {{ 5MMO q\A%y|a'78 ..Hb9 ]#__ 4D%  j'* ,,. :D*niH%dJ yQrc d}t}d}d}tjd}d}tjj |}d|j i}dD]} | r|} |j d i|} n|} |j dddddd |d} |j| tjts%|jtj| ||j| j|z|j| j||j| j!|t#|| | y) z-Ensures to_surface accepts an area arg/kwarg.r rrrr=arearNr/r2)rr rr!r rMrrtr>rrOrrr|rrrrV) rurrrrrrr!rrrRrs rtest_to_surface__area_paramz(MaskTypeTest.test_to_surface__area_param s/  ,#\\'2{{%$--/*' BJ!3,T__6v6 !6!__$dD&&.   ! !*fnn =  !<>PQ OOJ002]B C   Z335~ F   Z002D 9 j. A% Brctjd}tjd}d}tjj|d}|j }|ddz|ddzf}tj |t d }|j||j|d d d }|j|||j|j|t||||t||||y ) z$Ensures the default area is correct.rrrTrrr rrNrrrs rtest_to_surface__area_defaultz*MaskTypeTest.test_to_surface__area_default1 sg. U+  {{ 5MMO q\A%y|a'78 ..Hb9 ]#__ 4D%  j'* ,,. :D*niH%dJ yQrc^tjd}d}tjj|d}tj|}tjd}|j }|j |dddtjd fd |fd |j fd |fd |ff}t|}|D]f\} } |j| |j ||jdi|} |j| j|t|| |hy)z.Ensures to_surface accepts the correct kwargs.rr=TrrNNrr/ryellowrrrrNr2) r rr!r r>rrdictpoprrOrrV) rurRrr!rrr test_datarnamerIrs rtest_to_surface__kwargsz$MaskTypeTest.test_to_surface__kwargsI sg.{{40..& U+ \\^ '   6<<1 2  ( W\\^ , : &   i  BGD! JJt  LL '(262J   Z002D 9 j. A Brctjd}d}tjj|}tj|t d}tjd}|j ||j}|j |ddd|fdtjd fd |fd |fd f}t|}|D]u\} } |j| |jdi|} |j| tj|j| j|t|| |wy )zWEnsures to_surface accepts the correct kwargs when creating a surface. rr=rrrrrrrrr)rNNr2)r rr!r r>rrrrrrrtrOrrV) rurRrr!rsetsurface_colorrrrrrIrs r&test_to_surface__kwargs_create_surfacez3MaskTypeTest.test_to_surface__kwargs_create_surfaceh s& g.{{%^^D(B7 !<<.()!( .)   > * h/ 0 \ * : &  i  BGD! JJt (262J  ! !*fnn =   Z002D 9 j. A Brc ptjd}d}tjj|d}tj|}|j d|d|tj|d}|j |||j|j|t|||y)z2Ensures to_surface kwargs are not order dependent.bluerTrr/N)rrrrrr) r rr!r r>rrrOrrVrurRrr!rrs r)test_to_surface__kwargs_order_independentz6MaskTypeTest.test_to_surface__kwargs_order_independent sf-{{40..&__#- %  j'* ,,.5D*n=rc d}tjj|d}tjd}tj|}|j t 5|jddddddddd|j t 5|jdddd|ddd|j t 5|jddd|dddd|j t 5|jdd|ddddd|j t 5|jd|dddddd|j t 5|j|dddddddy#1swY xYw#1swYxYw#1swYxYw#1swYxYw#1swYnxYw#1swYyxYw)/Ensures to_surface detects invalid kwarg types.rTrgreenNr)r r!r rr>r|rr)rurr! invalid_surf invalid_colors r#test_to_surface__args_invalid_typesz0MaskTypeTest.test_to_surface__args_invalid_types s{{40||G, t,   y ) @ OOD$dD$ ? @  y ) C OOD$dM B C  y ) C OOD$mT B C  y ) B OOD$ dD A B  y ) B OOD,dD A B  y ) B OOL$dD A B B) @ @ C C C C B B B B B BsH#FF F=F'0F3#F?F FF$'F03F<?GcJd}tjj|}tj|tj|tj|tjdtjddd}dtjddtjdtjdd d}d}|D]E}t |}||||<|j t5|jd i|d d d Gy #1swYRxYw) rrrr/rrrrrr)rr rr)rrrr*rrrNr2) r r!r r>rrr|rr)rurr! valid_kwargsinvalid_kwargs kwarg_orderkwargrs r%test_to_surface__kwargs_invalid_typesz2MaskTypeTest.test_to_surface__kwargs_invalid_types s{{%~~d+ ..."NN40 W- ,,w/  $ ,,w/, F+ ..0   ! *E,'F*51F5M""9- *)&) * *  * * *s ;DD" ctjjd}dtjdi}|j t 5|j di|dddy#1swYyxYw)z/Ensures to_surface detects invalid kwarg names.r setcolourrNr2)r r!r rr|rr)rur!rs r$test_to_surface__kwargs_invalid_namez1MaskTypeTest.test_to_surface__kwargs_invalid_name s]{{'v||E23   y ) & DOO %f % & & &s A((A1cd}tjd}tjd}tjd}tjd}tjd}tj|td}|j }|j } |j || j |tj j|d } |} d d |fd |fd | fd|fd|fdf} g} t| }| D]Y\}}|| j||j||j || j| i|}t||| [y)z7Ensures to_surface accepts a combination of args/kwargsr=rrrrcyanrTrrrrrrrrN) r rr>rrrr!r rrJrrrV)rurrrunsetsurface_colorrrrrrr!rRrargsrrrrs r test_to_surface__args_and_kwargsz-MaskTypeTest.test_to_surface__args_and_kwargs sZ U+ !<<1#\\&1<<(\\&) ..x4\\^ ||~ (),-{{40)   : & \ *  " : &  i% BKD% E" JJt  LL '($9&9J j. A Brc&d}tjj|d}tj|td}tj d}dd|j ||ddd f}|D]!}|j| }t|||#y ) zrrmap_rgbrrV)rurr!rrR test_colorsrrs r'test_to_surface__valid_setcolor_formatsz4MaskTypeTest.test_to_surface__valid_setcolor_formats s{{40..x4g.   OON +     $ BH(;J j. A Brc"d}tjj|}tj|td}tj d}dd|j ||dddf}|D]!}|j|}t|||#y ) z>Ensures to_surface handles valid unsetcolor formats correctly.r=rrrrrrrNr)rurr!rrRrrrs r)test_to_surface__valid_unsetcolor_formatsz6MaskTypeTest.test_to_surface__valid_unsetcolor_formats- s{{%..x4g.   OON +     & BJJ?J j. A Brctjjd}dD]2}|jt5|j |ddd4tj dtjddfD]2}|jt5|j |ddd4y#1swYxYw#1swYKxYw)z>Ensures to_surface handles invalid setcolor formats correctly.r=z green colorz #00FF00FF0 0x00FF00FF0rrNrr皙?r r!r r|r}rr>r)rur!rs r)test_to_surface__invalid_setcolor_formatsz6MaskTypeTest.test_to_surface__invalid_setcolor_formatsB s{{'L 3H"":. 32 3 3 3 /V1DcJ 3H""9- 32 3 3 3 3 3 3 3B8C8C C ctjjd}dD]2}|jt5|j |ddd4tj dtjddfD]2}|jt5|j |ddd4y#1swYxYw#1swYKxYw)z@Ensures to_surface handles invalid unsetcolor formats correctly.r=rrNrrrr)rur!rs r+test_to_surface__invalid_unsetcolor_formatsz8MaskTypeTest.test_to_surface__invalid_unsetcolor_formatsN s{{'N 7J"":. 7:6 7 7 7">>&16;;v3FL 7J""9- 7:6 7 7 7 7 7 7 7r c tjd}tjjdd}dddgt dddtj ddf}|D]!}|j | }t|||#y ) z8Ensures to_surface handles valid dest formats correctly.rrUTrr/r)rrrrrrN)r rr!r rr'rrVrurRr!destsrrs r#test_to_surface__valid_dest_formatsz0MaskTypeTest.test_to_surface__valid_dest_formatsZ sg.{{T2  F AqM  KK )   BDd3J j. A Brctjjd}ddddhdditf}|D]2}|j t 5|j |ddd4y#1swY?xYw)z:Ensures to_surface handles invalid dest formats correctly.rUrrrrr N)r r!r r'r|rr)rur! invalid_destsrs r%test_to_surface__invalid_dest_formatsz2MaskTypeTest.test_to_surface__invalid_dest_formatsk s{{{'   F F   " +D""9- +T* + + + + +s A''A0 c>tjd}tjjdd}tjddtjddtjddf}|D]!}|j | }t |||#y ) z?Ensures to_surface correctly handles negative sized dest rects.rrUTrr/)ra)rra)rrr Nr rr!r r'rrVrs r)test_to_surface__negative_sized_dest_rectz6MaskTypeTest.test_to_surface__negative_sized_dest_rectz sg.{{T2 KK * KK * KK +   BDd3J j. A Brc>tjd}tjjdd}tjddtjddtjddf}|D]!}|j |}t |||#y ) z;Ensures to_surface correctly handles zero sized dest rects.rrUTrr/rrr Nrrs r%test_to_surface__zero_sized_dest_rectz2MaskTypeTest.test_to_surface__zero_sized_dest_rect sg.{{T2 KK ( KK ( KK '   BDd3J j. A Brcd}tjd}tjd}tj|}tjj |d}d}d}|d|d |d|d f||f|t |ft ||ft |t |f|d|d |d|d g||g|t |gt ||gt |t |gtj ||f }|D]W} |j|tj | } |j|| } t|| || t|| || Yy ) z8Ensures to_surface handles valid area formats correctly.rUrrTrr/rrrrN) r rr>r!r listr'rrrVr[) rurrrRrr!area_pos area_sizeareasrrSrs r#test_to_surface__valid_area_formatsz0MaskTypeTest.test_to_surface__valid_area_formats s^ U+ g...&{{40 a[(1+y|Yq\ B y ! tI ' (^Y ' (^T)_ - a[(1+y|Yq\ B y ! tI ' (^Y ' (^T)_ - KK) ,   VD LL ' D)Itrrr rrrr0zinvalid area argumentrN)r r!r r'assertRaisesRegexrr)rur! invalid_areasrunused_to_surfaces r%test_to_surface__invalid_area_formatsz2MaskTypeTest.test_to_surface__invalid_area_formats s{{'      aL   " ?D'' 3JK ?$(OOO$>! ? ? ? ? ?s A,,A5 cd}tjd}tjd}tj|}tjj |}|j dtj ddtj ddtj d d f}|D]B}|j||j|| }t||||t||||Dy ) z?Ensures to_surface correctly handles negative sized area rects.rUrrr/rfrirhrnrgrlrN) r rr>r!r rr'rrrVr[) rurrrRrr!rrrs r)test_to_surface__negative_sized_area_rectz6MaskTypeTest.test_to_surface__negative_sized_area_rect s U+ g...&{{% F KK ( KK ( KK )   QD LL 'tr!r r'rrrV)rurrRrr!rrrs r%test_to_surface__zero_sized_area_rectz2MaskTypeTest.test_to_surface__zero_sized_area_rect se,..&{{40 KK ' KK ' KK '   BD LL (trrr!r rrrtrrOrrr|rrrrV)rurrrrrrrrrrrrrrr!setsurface_paramunsetsurface_paramsetcolor_paramunsetcolor_param dest_paramrRrs r8test_to_surface__default_surface_with_param_combinationszEMaskTypeTest.test_to_surface__default_surface_with_param_combinations ss  ,!<<1#\\&1<<(\\&) ^^D-H !( (),-   "1 VD;;##Dt#4D&0$6- V '7|$+7*>*V&-?F>*+3T*:'V-;z*1;T0B$V,3CF<0/3Tl!V #-#5$*JJvt$<5?F6N#''7'C9I)7)C9A9N'9'E9K)9)E9C9N-rrr!r rrrrrOrrr|rrrrV)rurrrrrrrrrrrrrrrr!r+r,r-r.r/rRrs r0test_to_surface__surface_with_param_combinationsz=MaskTypeTest.test_to_surface__surface_with_param_combinationsO s  U+ !<<1#\\&1<<(\\&) ..}nE\\^ ||~ (),-   "2 VD;;##Dt#4D&0$6. V '7|$+7*>+V&-?F>*+3T*:(V-;z*1;T0B%V,3CF<0#LL7/3Tl!V #-#5$*JJvt$<5?F6N#''7'C9I)7)C9A9F'9'E9K)9)E9C9F-. .Rq%Avq:QRQQFRFR#&r c3FK|]}tD]}||f ywrFrr7s rr7zCMaskTypeTest.test_to_surface__set_and_unset_bits.. %L1eFmLALLrrrN)r rr!r rMr>r rr#rrrrNrYrrOrP)rudefault_setcolorrr8rr!rrrrUrrmask_posrRr9s @r#test_to_surface__set_and_unset_bitsz0MaskTypeTest.test_to_surface__set_and_unset_bits sq"<<0#\\'2'' v{{%MMO ..& U+ SE%LR C KK  '' D $I  LL 't? --c2%2N[[*%5N%7N  !2!23!7$PSU V    # rc dx\}}tjj|}|j}tj|}tj d}|j }tj d}|j||j } tj d} | j| fdt|DD]} |j| |jD]} | |_ dD]} |j|| r|j|| || dd}n|j|| || }|jfd t|DD]c} | d | d z | d | d z f}|j| s|}n|j|r|}n| }|j!|j| |e|j#y) zyEnsures that to_surface works correctly with with set/unset bits when using setsurface and unsetsurface. r4rrrc3PK|]}t|dzdD]}||f ywr6rr7s rr7z`MaskTypeTest.test_to_surface__set_and_unset_bits_with_setsurface_unsetsurface.. r8r9rN)rrrrr)rrrc3FK|]}tD]}||f ywrFrr7s rr7z`MaskTypeTest.test_to_surface__set_and_unset_bits_with_setsurface_unsetsurface.. s%Pq%-PQQFPFPrrr)r r!r rMr>rrrr rr#rrrNrYrrOrP)rur8rr!rrrrrrrrUrdisable_color_paramsrr=rRr9s @r@test_to_surface__set_and_unset_bits_with_setsurface_unsetsurfacezMMaskTypeTest.test_to_surface__set_and_unset_bits_with_setsurface_unsetsurface s (' v{{%MMO ..& U+ \\^ !<<0()||~ #\\&1,-SE%LR C KK  ''' $D $I  )6! $$ ]+'!%!#-%1!%#' "1"J"&!#-%1 "1"J!PE%LP MC #Aa 0#a&472BCH$11#6)6X.)9);$$Z%6%6s%;^L M!!#C! $ ' $rctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j ||r|n|} |j |} |j| ||j| j|t|| | y) zEnsures that surfaces narrower than the mask work correctly. For this test the surface's width is less than the mask's width. rrr4rrrrN r rr>r!r rrrrOrrV) rur<rr9 narrow_sizerrrr!rRrs r+test_to_surface__surface_narrower_than_maskz8MaskTypeTest.test_to_surface__surface_narrower_than_mask s "<<0#\\'2  ..- U+ ! BD;;##ID#9D LL '15-;MN1J MM*g .   Z002K @ j. A BrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| ||t|| ||t|| |y ) zEnsures that setsurfaces narrower than the mask work correctly. For this test the setsurface's width is less than the mask's width. rrr4rErrrrrNr rr>rrrMr!r rrtrOrrVr[) rur<rr9rHrrsetsurface_rectrr!rs r.test_to_surface__setsurface_narrower_than_maskz;MaskTypeTest.test_to_surface__setsurface_narrower_than_mask$s "<<0#\\'2  ^^K2> !<<.()$--/! JD;;##ID#9DJ?J  ! !*fnn =   Z002I >#D*6FX-*&6$D*6HI JrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| |t|| ||t|| ||y ) zEnsures that unsetsurfaces narrower than the mask work correctly. For this test the unsetsurface's width is less than the mask's width. rrr4rErrrrrNrL) rur<rr9rHrrunsetsurface_rectrr!rs r0test_to_surface__unsetsurface_narrower_than_maskz=MaskTypeTest.test_to_surface__unsetsurface_narrower_than_maskDs  "<<0#\\'2  ~~k8R@ #\\%0,-(113! D;;##ID#9DlCJ  ! !*fnn =   Z002I >#D*6FG#*&8:K.*&8:K rcd}d}d}tj|td}tjd}|j ||j }dD]}tj j||}|j|dd } |j| tj|j| j||rt|| ||t|| ||t|| |y) zEnsures that setsurfaces narrower than the mask work correctly when setcolor and unsetcolor are set to None. For this test the setsurface's width is less than the mask's width. rr4rErrrrNrrrr r>rrrrMr!r rrtrOrrVr[) rurr9rHrrrMrr!rs r>test_to_surface__setsurface_narrower_than_mask_and_colors_nonezKMaskTypeTest.test_to_surface__setsurface_narrower_than_mask_and_colors_nonefs !-  ^^K2> !<<.()$--/! MD;;##ID#9D%)J  ! !*fnn =   Z002I >#D*6FX-*&;_$D*6KL# Mrcd}d}d}tj|td}tjd}|j ||j }dD]}tj j||}|j|dd } |j| tj|j| j||rt|| |t|| ||t|| ||y) zEnsures that unsetsurfaces narrower than the mask work correctly when setcolor and unsetcolor are set to None. For this test the unsetsurface's width is less than the mask's width. rr4rErrrrNrrrrU) rurr9rHrrrQrr!rs r@test_to_surface__unsetsurface_narrower_than_mask_and_colors_nonezMMaskTypeTest.test_to_surface__unsetsurface_narrower_than_mask_and_colors_nones !-  ~~k8R@ #\\%0,-(113! D;;##ID#9D)DT)J  ! !*fnn =   Z002I >#D*6KL#*&8:K.*&;=N# rctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j } |j ||r|n|} |j|} |j| ||j| j|t|| | | t|| || y) zEnsures that surfaces wider than the mask work correctly. For this test the surface's width is greater than the mask's width. rrrFrqr\rrrN r rr>r!r rMrrrrOrrVr[) rur<rr9 wide_sizerrrr!rrRrs r(test_to_surface__surface_wider_than_maskz5MaskTypeTest.test_to_surface__surface_wider_than_mask "<<0#\\'2  ..+ U+ ! VD;;##ID#9D I LL '15-;MN1J MM*g .   Z002I > j.) L )$ M9 U Vrctjd}d}d}tj|td}tjd}|j |dD]}tj j ||}|r|n|}|j|} |j| tj|j| j|t|| |y ) zEnsures that setsurfaces wider than the mask work correctly. For this test the setsurface's width is greater than the mask's width. rr[r]rrrrrKN r rr>rrr!r rrtrOrrV) rurr9r_rrrr!rRrs r+test_to_surface__setsurface_wider_than_maskz8MaskTypeTest.test_to_surface__setsurface_wider_than_masks $\\'2  ^^Ix< !<<.()! BD;;##ID#9D15-;MNJ?J  ! !*fnn =   Z002I > j. A Brctjd}d}d}tj|td}tjd}|j |dD]}tj j ||}|r|n|}|j|} |j| tj|j| j|t|| |y ) zEnsures that unsetsurfaces wider than the mask work correctly. For this test the unsetsurface's width is greater than the mask's width. rr[r]rrrrrPNrc) rur<r9r_rrrr!rRrs r-test_to_surface__unsetsurface_wider_than_maskz:MaskTypeTest.test_to_surface__unsetsurface_wider_than_mask "<<0  ~~i2> #\\%0,-! BD;;##ID#9D15-;MNlCJ  ! !*fnn =   Z002I > j. A Brctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j ||r|n|} |j |} |j| ||j| j|t|| | y) zEnsures that surfaces shorter than the mask work correctly. For this test the surface's height is less than the mask's height. rrrarqrarFrrrNrG) rur<rr9 short_sizerrrr!rRrs r*test_to_surface__surface_shorter_than_maskz7MaskTypeTest.test_to_surface__surface_shorter_than_masks "<<0#\\'2  .., U+ ! BD;;##ID#9D LL '15-;MN1J MM*g .   Z002J ? j. A BrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| ||t|| ||t|| |y ) zEnsures that setsurfaces shorter than the mask work correctly. For this test the setsurface's height is less than the mask's height. rrrirjrrrrrKNrL) rur<rr9rkrrrMrr!rs r-test_to_surface__setsurface_shorter_than_maskz:MaskTypeTest.test_to_surface__setsurface_shorter_than_mask s "<<0#\\'2  ^^J"= !<<.()$--/! JD;;##ID#9DJ?J  ! !*fnn =   Z002I >#D*6FX-*&6$D*6HI JrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| |t|| ||t|| ||y ) zEnsures that unsetsurfaces shorter than the mask work correctly. For this test the unsetsurface's height is less than the mask's height. rrrirjrrrrrPNrL) rur<rr9rkrrrQrr!rs r/test_to_surface__unsetsurface_shorter_than_maskz#D*6FG#*&8:K.*&8:K rcd}d}d}tj|td}tjd}|j ||j }dD]}tj j||}|j|dd } |j| tj|j| j||rt|| ||t|| ||t|| |y) zEnsures that setsurfaces shorter than the mask work correctly when setcolor and unsetcolor are set to None. For this test the setsurface's height is less than the mask's height. rrirjrrrrNrTrU) rurr9rkrrrMrr!rs r=test_to_surface__setsurface_shorter_than_mask_and_colors_nonezJMaskTypeTest.test_to_surface__setsurface_shorter_than_mask_and_colors_noneOs !-  ^^J"= !<<.()$--/! MD;;##ID#9D%)J  ! !*fnn =   Z002I >#D*6FX-*&;_$D*6KL# Mrcd}d}d}tj|td}tjd}|j ||j }dD]}tj j||}|j|dd } |j| tj|j| j||rt|| |t|| ||t|| ||y) zEnsures that unsetsurfaces shorter than the mask work correctly when setcolor and unsetcolor are set to None. For this test the unsetsurface's height is less than the mask's height. rrirjrrrrNrXrU) rurr9rkrrrQrr!rs r?test_to_surface__unsetsurface_shorter_than_mask_and_colors_nonezLMaskTypeTest.test_to_surface__unsetsurface_shorter_than_mask_and_colors_noneqs !-  ~~j(B? #\\%0,-(113! D;;##ID#9D)DT)J  ! !*fnn =   Z002I >#D*6KL#*&8:K.*&;=N# rctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j } |j ||r|n|} |j|} |j| ||j| j|t|| | | t|| || y) zEnsures that surfaces taller than the mask work correctly. For this test the surface's height is greater than the mask's height. rrrjrirrrNr^) rur<rr9 tall_sizerrrr!rrRrs r)test_to_surface__surface_taller_than_maskz6MaskTypeTest.test_to_surface__surface_taller_than_maskrarctjd}d}d}tj|td}tjd}|j |dD]}tj j ||}|r|n|}|j|} |j| tj|j| j|t|| |y ) zEnsures that setsurfaces taller than the mask work correctly. For this test the setsurface's height is greater than the mask's height. rrjrirrrrrKNrc) rurr9rvrrrr!rRrs r,test_to_surface__setsurface_taller_than_maskz9MaskTypeTest.test_to_surface__setsurface_taller_than_masks $\\'2  ^^Ix< !<<.()! BD;;##ID#9D15-;MNJ?J  ! !*fnn =   Z002I > j. A Brctjd}d}d}tj|td}tjd}|j |dD]}tj j ||}|r|n|}|j|} |j| tj|j| j|t|| |y ) zEnsures that unsetsurfaces taller than the mask work correctly. For this test the unsetsurface's height is greater than the mask's height. rrjrirrrrrPNrc) rur<r9rvrrrr!rRrs r.test_to_surface__unsetsurface_taller_than_maskz;MaskTypeTest.test_to_surface__unsetsurface_taller_than_maskrgrctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j } |j ||r|n|} |j|} |j| ||j| j|t|| | | t|| || y) zEnsures that surfaces wider and taller than the mask work correctly. For this test the surface's width is greater than the mask's width and the surface's height is greater than the mask's height. rrrFr]rrrNr^) rur<rr9wide_tall_sizerrrr!rrRrs r3test_to_surface__surface_wider_and_taller_than_maskz@MaskTypeTest.test_to_surface__surface_wider_and_taller_than_masks "<<0#\\'2 !..0 U+ ! VD;;##ID#9D I LL '15-;MN1J MM*g .   Z002N C j.) L )$ M9 U Vrctjd}d}d}tj|td}tjd}|j |dD]}tj j ||}|r|n|}|j|} |j| tj|j| j|t|| |y ) zEnsures that setsurfaces wider and taller than the mask work correctly. For this test the setsurface's width is greater than the mask's width and the setsurface's height is greater than the mask's height. rr}r]rrrrrKNrc) rurr9rrrrr!rRrs r6test_to_surface__setsurface_wider_and_taller_than_maskzCMaskTypeTest.test_to_surface__setsurface_wider_and_taller_than_masks$\\'2 !^^NHbA !<<.()! BD;;##ID#9D15-;MNJ?J  ! !*fnn =   Z002I > j. A Brctjd}d}d}tj|td}tjd}|j |dD]}tj j ||}|r|n|}|j|} |j| tj|j| j|t|| |y ) zEnsures that unsetsurfaces wider and taller than the mask work correctly. For this test the unsetsurface's width is greater than the mask's width and the unsetsurface's height is greater than the mask's height. rr}r]rrrrrPNrc) rur<r9rrrrr!rRrs r8test_to_surface__unsetsurface_wider_and_taller_than_maskzEMaskTypeTest.test_to_surface__unsetsurface_wider_and_taller_than_masks"<<0 !~~nhC #\\%0,-! BD;;##ID#9D15-;MNlCJ  ! !*fnn =   Z002I > j. A Brctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j } |j ||r|n|} |j|} |j| ||j| j|t|| | | t|| || y) zEnsures that surfaces wider and shorter than the mask work correctly. For this test the surface's width is greater than the mask's width and the surface's height is less than the mask's height. rrr)rrFrrrNr^) rur<rr9wide_short_sizerrrr!rrRrs r4test_to_surface__surface_wider_and_shorter_than_maskzAMaskTypeTest.test_to_surface__surface_wider_and_shorter_than_mask,s"<<0#\\'2 !..1 U+ ! VD;;##ID#9D I LL '15-;MN1J MM*g .   Z002O D j.) L )$ M9 U VrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| ||t|| ||t|| |y ) zEnsures that setsurfaces wider and shorter than the mask work correctly. For this test the setsurface's width is greater than the mask's width and the setsurface's height is less than the mask's height. rrrrjrrrrrKNrL) rur<rr9rrrrMrr!rs r7test_to_surface__setsurface_wider_and_shorter_than_maskzDMaskTypeTest.test_to_surface__setsurface_wider_and_shorter_than_maskHs"<<0#\\'2 !^^OXrB !<<.()$--/! JD;;##ID#9DJ?J  ! !*fnn =   Z002I >#D*6FX-*&6$D*6HI JrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| |t|| ||t|| ||y ) zEnsures that unsetsurfaces wider and shorter than the mask work correctly. For this test the unsetsurface's width is greater than the mask's width and the unsetsurface's height is less than the mask's height. rrrrjrrrrrPNrL) rur<rr9rrrrQrr!rs r9test_to_surface__unsetsurface_wider_and_shorter_than_maskzFMaskTypeTest.test_to_surface__unsetsurface_wider_and_shorter_than_maskjs "<<0#\\'2 !~~oxD #\\%0,-(113! D;;##ID#9DlCJ  ! !*fnn =   Z002I >#D*6FG#*&8:K.*&8:K rctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j } |j ||r|n|} |j|} |j| ||j| j|t|| | | t|| || y) zEnsures that surfaces narrower and taller than the mask work correctly. For this test the surface's width is less than the mask's width and the surface's height is greater than the mask's height. rrrar~r[rrrNr^) rur<rr9narrow_tall_sizerrrr!rrRrs r6test_to_surface__surface_narrower_and_taller_than_maskzCMaskTypeTest.test_to_surface__surface_narrower_and_taller_than_masks"<<0#\\'2 "..!12 U+ ! VD;;##ID#9D I LL '15-;MN1J MM*g .   Z0024D E j.) L )$ M9 U VrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| ||t|| ||t|| |y ) zEnsures that setsurfaces narrower and taller than the mask work correctly. For this test the setsurface's width is less than the mask's width and the setsurface's height is greater than the mask's height. rrrr[rrrrrKNrL) rur<rr9rrrrMrr!rs r9test_to_surface__setsurface_narrower_and_taller_than_maskzFMaskTypeTest.test_to_surface__setsurface_narrower_and_taller_than_masks"<<0#\\'2 "^^$4hC !<<.()$--/! JD;;##ID#9DJ?J  ! !*fnn =   Z002I >#D*6FX-*&6$D*6HI JrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| |t|| ||t|| ||y ) zEnsures that unsetsurfaces narrower and taller than the mask work correctly. For this test the unsetsurface's width is less than the mask's width and the unsetsurface's height is greater than the mask's height. rrrr[rrrrrPNrL) rur<rr9rrrrQrr!rs r;test_to_surface__unsetsurface_narrower_and_taller_than_maskzHMaskTypeTest.test_to_surface__unsetsurface_narrower_and_taller_than_masks "<<0#\\'2 "~~&6"E #\\%0,-(113! D;;##ID#9DlCJ  ! !*fnn =   Z002I >#D*6FG#*&8:K.*&8:K rctjd}tjd}d}d}tj|}tjd}dD]}tjj ||}|j } |j ||r|n|} |j|} |j| ||j| j|t|| | | t|| || y) zEnsures that surfaces narrower and shorter than the mask work correctly. For this test the surface's width is less than the mask's width and the surface's height is less than the mask's height. rrrar[rrrNr^) rur<rr9narrow_short_sizerrrr!rrRrs r7test_to_surface__surface_narrower_and_shorter_than_maskzDMaskTypeTest.test_to_surface__surface_narrower_and_shorter_than_masks"<<0#\\'2 #..!23 U+ ! VD;;##ID#9D I LL '15-;MN1J MM*g .   Z0024E F j.) L )$ M9 U VrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| ||t|| ||t|| |y ) zEnsures that setsurfaces narrower and shorter than the mask work correctly. For this test the setsurface's width is less than the mask's width and the setsurface's height is less than the mask's height. rrrr[rrrrrKNrL) rur<rr9rrrrMrr!rs r:test_to_surface__setsurface_narrower_and_shorter_than_maskzGMaskTypeTest.test_to_surface__setsurface_narrower_and_shorter_than_mask s"<<0#\\'2 #^^$5xD !<<.()$--/! JD;;##ID#9DJ?J  ! !*fnn =   Z002I >#D*6FX-*&6$D*6HI JrcNtjd}tjd}d}d}tj|td}tjd}|j ||j }dD]}tj j||} | j| } |j| tj|j| j||rt|| |t|| ||t|| ||y ) zEnsures that unsetsurfaces narrower and shorter than the mask work correctly. For this test the unsetsurface's width is less than the mask's width and the unsetsurface's height is less than the mask's height. rrrr[rrrrrPNrL) rur<rr9rrrrQrr!rs r#D*6FG#*&8:K.*&8:K rcTtjd}tjd}tjd}tjd}tjd}d}d}d}d } tj|} tj|} tj| } | j|| j|| j|| j | j } | j }tj j |d }|j }tjd d |j_fdtjjDD]}|j|d|j| | | }|j|| |j|j!||j#fdtjDD]u}|j%|s|}n>|j'|r| j%|r|}n|}n|j%|r|}n|}|j|j'||w|j)y)zBEnsures that all the surface parameters can be of different sizes.rrrrr)rar\)rq)rwr~)rTrr/rc3pK|]-}tjjD]}||f /ywrF)r rr)r3rr unfilled_rects rr7zWMaskTypeTest.test_to_surface__all_surfaces_different_sizes_than_mask..rsA =??MOO< F  s36rc3ZK|]"}tjD]}||f $ywrF)r r)r3rr surface_rects rr7zWMaskTypeTest.test_to_surface__all_surfaces_different_sizes_than_mask..s5 5;P 67QF  s(+N)r rr>rrMr!r r'rr rrrrrrOrrNrYrrP)rur<rrrrr9 surface_sizesetsurface_sizeunsetsurface_sizerrrrMrQr!rrUrrRrrs @@r7test_to_surface__all_surfaces_different_sizes_than_maskzDMaskTypeTest.test_to_surface__all_surfaces_different_sizes_than_maskRsE"<<0#\\'2 U+ !<<0#\\&1   $...^^O4 ~~&78  ]#(),-'') $--/(113{{ 5MMO  FF3 (//  =??MOO<  C KKQ   __Wj,G  j'* ,,. =  !,..1  EC))#.!.S!"//4%5N%5N%11#6%7N%7N   Z..s3^ D% E( rc d}tj||f}|j}|j}tjd}tjd}tjd}dt | |dzDdt | |dzDdt | |dzDd t | |dzDf}d D]} tj j||f| } | r|n|} |D]a} | D]Z} | |_|j|}|j|| j|| }t||| |t||||\cy )zBEnsures dest values can be different locations on/off the surface.rrrrc3$K|]}|df ywr1r2r3ss rr7z?MaskTypeTest.test_to_surface__dest_locations.. 4aV 4rc3$K|]}d|f ywr1r2rs rr7z?MaskTypeTest.test_to_surface__dest_locations..rrc3$K|]}||f ywrFr2rs rr7z?MaskTypeTest.test_to_surface__dest_locations..rrc3&K|] }| |f ywrFr2rs rr7z?MaskTypeTest.test_to_surface__dest_locations.. 5qb!W 5rrr N)r r>rMrrr r!r rrLrrrVr[)ruSIDErr dest_rectrr<r directionsrr!rR directionrUrMrs rtest_to_surface__dest_locationsz,MaskTypeTest.test_to_surface__dest_locationssh..$.'')  %%'  U+ !<<0#\\'2 5UD5$(3 4 4UD5$(3 4 4UD5$(3 4 5eTE4!84 5  " D;;##T4Lt#>,#?LLL/!%y!IJ'j.,W1j-    rc d}tj||f}tjd}tjd}tjd}dt| |dzDdt| |dzDdt| |dzDd t| |dzDf}d D]}tjj ||f| }|j } | j} |r|n|} |D]h} | D]a} | | _| j| }d |_|j||j|| }t||| |t||||cjy)z>Ensures area rects can be different locations on/off the mask.rrrrc3$K|]}|df ywr1r2rs rr7z?MaskTypeTest.test_to_surface__area_locations..rrrc3$K|]}d|f ywr1r2rs rr7z?MaskTypeTest.test_to_surface__area_locations..rrc3$K|]}||f ywrFr2rs rr7z?MaskTypeTest.test_to_surface__area_locations..rrc3&K|] }| |f ywrFr2rs rr7z?MaskTypeTest.test_to_surface__area_locations..rrrrr/rN)r r>rr r!r rMrrrLrrrVr[)rurrrr<rrrr!rrSrRrrUrMrs rtest_to_surface__area_locationsz,MaskTypeTest.test_to_surface__area_locationssm..$. U+ !<<0#\\'2 5UD5$(3 4 4UD5$(3 4 4UD5$(3 4 5eTE4!84 5  " D;;##T4Lt#>)#.rrrc3$K|]}d|f ywr1r2rs rr7zHMaskTypeTest.test_to_surface__dest_and_area_locations..rrc3$K|]}||f ywrFr2rs rr7zHMaskTypeTest.test_to_surface__dest_and_area_locations..rrc3&K|] }| |f ywrFr2rs rr7zHMaskTypeTest.test_to_surface__dest_and_area_locations..rrr rr)rrN)r r>rMrrr rr!r rrLrrrVr[)rurrrrrr<rdest_directionsarea_positionsrr!rrSrRdest_directiondest_posrarea_overlap_rectdest_overlap_rectrs r(test_to_surface__dest_and_area_locationsz5MaskTypeTest.test_to_surface__dest_and_area_locationss ..$.'')  %%'  U+ !<<0#\\'2 5UD5$(3 4 4UD5$(3 4 4UD5$(3 4 5eTE4!84 5  oa01! D;;##T4Lt#O6 *m=N   rc d}dddd|dz |dz f|dz |f||dz f||f|dz|f||dzf|dz|dzff }tj||f}tjd}tjd}tjd }d D]}tjj ||f| }|j } |r|n|} |D]} tj d| } |jD]a} | | _| j| }d|_|j||j|| }t||| |t||||cy )z*Ensures area rects can be different sizes.rr/rfrhrgrrrrrrrN)r r>rr!r rMr'r#rrLrrrVr[)rurSIZESrrr<rrr!rrRrrSrUrMrs rtest_to_surface__area_sizesz(MaskTypeTest.test_to_surface__area_sizess     AXtax AXt  4!8  4L AXt  4!8  AXtax   ..$. U+ !<<0#\\'2! D;;##T4Lt#>)#rrr!r rrtrOrrV) rurrrrrrr!rRrs r%test_to_surface__surface_color_alphasz2MaskTypeTest.test_to_surface__surface_color_alphasBs!<<0#\\&1"^^DC ~~d("E (),-! BD;;##Dt#4D15-;MN%L)J  ! !*fnn =   Z002D 9 j. A Brcd}tjd}d|_tjd}d|_dD]}tjj ||}|r|n|}|j ||}|j |tj|j|j|t|||y ) z;Ensures the setcolor/unsetcolor alpha values are respected.rr#rrr)rrN) r rr4r!r rrtr>rOrrV)rurrrrr!rRrs rtest_to_surface__color_alphasz*MaskTypeTest.test_to_surface__color_alphas\s<<( \\&)  ! BD;;##Dt#4D)-X:N(zRJ  ! !*fnn =   Z002D 9 j. A Brcd}tjd}tjd}tjd}dD]'}tj||}tj||}tj||}|j||j||j|dD]} tjj || } | r|j d n|j d } | j|||} |j| tj|j| j|t|| | *y ) zAEnsures to_surface works correctly with supported surface depths.rrrrr~rrrrrr/N) r rr>rr!r rrrtrOrrV) rurrrrrrrrrr!rRrs rtest_to_surface__depthsz$MaskTypeTest.test_to_surface__depthsns7 U+ !<<0#\\&1$ FEnnT7GE:J!>>$err|r}r) rurrrrr! test_depthsdepthsrrrs r!test_to_surface__different_depthsz.MaskTypeTest.test_to_surface__different_depthss U+ !<<0#\\&1{{%  " CFnnT;GF1I>J!>>$fQi@L LL ' OO, -   0 1"":. C\B C C C C Cs >DD& cd}tjd}tjd}tjj|}d}|D]\}}tj||}tj||} |j || j ||j t5|j|| dddy#1swYxYw)zmEnsures an exception is raised when surfaces have different depths than the created surface. rrr))r~r~)rr)rr)rr)rr~)rr)rrrrrrNr) rurrrr!r set_depth unset_depthrrs r7test_to_surface__different_depths_with_created_surfaceszDMaskTypeTest.test_to_surface__different_depths_with_created_surfacess!<<0#\\&1{{%  '2 R "I{I>J!>>$kBL OO, -   0 1"":. R:LQ R R R R Rs ;CC# cd}tjd}tjd}tjd}dD]@}dtfD]1}tj|||}tj|||}tj|||} |j ||j || j |dD]} tj j || } | r|n|} | j||| } |j| tj|j| j|t|| | |s|j| j|z4Cy ) zHEnsures to_surface works correctly when the SRCALPHA flag is set or not.rrrrrrrrrNr rrr>rr!r rrtrOrrVr|r)rurrrrrrrrrrr!rRrs rtest_to_surface__same_srcalphasz,MaskTypeTest.test_to_surface__same_srcalphassT U+ !<<0#\\&1 HEX H ..U%H#^^DUK %~~d%uM  ]+ 01!!"45) HD!;;++Dt+(FG H H Hrcld}tjd}tjd}t}tj||d}tj||d}|j ||j |dD]}tj j ||}|r|n|} |j||} |j| tj|j| j|t|| | |j| j|zy ) zEnsures to_surface works correctly when it creates a surface and the SRCALPHA flag is set on both setsurface and unsetsurface. rrrrrrrrNr) rurrrexpected_flagsrrrr!rRrs r5test_to_surface__same_srcalphas_with_created_surfaceszBMaskTypeTest.test_to_surface__same_srcalphas_with_created_surfacess!<<0#\\&1!^^DbI ~~d.K (),-! ED;;##Dt#4D15-;MN%L)J  ! !*fnn =   Z002D 9 j. A OOJ002^C D Ercd}tjd}tjd}tjd}tjj|}tddfttdfdttfddtff}dD]}|D]}tj ||d|} tj ||d|} tj ||d |} | j || j || j ||jt5|j| | | d d d y #1swYxYw) zdEnsures an exception is raised when surfaces have different SRCALPHA flag settings. rrrrrrrrr N r rr!r rr>rr|r}r) rurrrrr! test_flagsrrrrrs r$test_to_surface__different_srcalphasz1MaskTypeTest.test_to_surface__different_srcalphassE U+ !<<0#\\&1{{%q!  x # ( # 8    GE# G ..U1XUK#^^DaN %~~d%(%P  ]+ 01!!"45&&z2GOOGZFGG G GGGs .EE cd}tjd}tjd}tjj|}dD]}dtdfdtffD]}tj ||d|}tj ||d|}|j ||j ||jt5|j|| d d d y #1swYxYw) z}Ensures an exception is raised when surfaces have different SRCALPHA flag settings than the created surface. rrrrr/rrrrNr) rurrrr!rrrrs r:test_to_surface__different_srcalphas_with_created_surfaceszGMaskTypeTest.test_to_surface__different_srcalphas_with_created_surfaces's!<<0#\\&1{{% VE!8Q-!X? V#^^DaN %~~d%(%P  01!!"45&&z2VOOz OUVV V VVVs C33C< ch tjd}tjd}dx\ }}tj|td}tjd}dD]}tjj ||}|j } |r|n|} fdt|DD]{} |j|| | _ |j|| } |j| ||j| j|t|| | | t|| || }y ) zwEnsures dest values on the surface work correctly when using the defaults for setcolor and unsetcolor. rrrrwrrrrc3FK|]}tD]}||f ywrFrrs rr7z@MaskTypeTest.test_to_surface__dest_on_surface..M%MAe M1!QMMrr N)r rr>rr!r rMr rrrrrOrrVr[)rur<rr9rrrrr!rrRrrr8s @r test_to_surface__dest_on_surfacez-MaskTypeTest.test_to_surface__dest_on_surface=s"<<0#\\'2%% v..x4 U+ ! D;;##Dt#4D I15-;MNNvM  ]+$( !!__W4_@  j'2  !4!4!6=#D*niP-*mY   rc Tdx\}}tj|td}tjd}|j }tjd}|j ||j }tjd}|j ||||dd} t | } | jdd D]} tjj|| } | j} | r|n|}fd t|DD]}|| _ d D]~}|j ||r| n| }||d <| jd i|}|j|||j|j!|t#|||| t%|||| y)zjEnsures dest values on the surface work correctly when using setsurface and unsetsurface. rrrrrNrrrr)rN)rNrrc3FK|]}tD]}||f ywrFrrs rr7z]MaskTypeTest.test_to_surface__dest_on_surface_with_setsurface_unsetsurface..|rrrr2)r r>rrrrrupdater!r rMr rrrrOrrVr[)rur9rrrrrrrr color_kwargsrr!rrRruse_color_params test_kwargsrr8s @r=test_to_surface__dest_on_surface_with_setsurface_unsetsurfacezJMaskTypeTest.test_to_surface__dest_on_surface_with_setsurface_unsetsurfaceZs &% v..x4 U+ \\^ !<<0()||~ #\\&1,- $(  F| FG! D;;##Dt#4D I15-;MNNvM $( !(5 $LL/2B,K*.K'!0!?;!?JMM*g6$$Z%8%8%:DA'j.)T1j-    rctjd}tjd}dx\}}}tj|td}tjd}| | f| dfd| fg}|j t |j dD]} tjj|| } | j } | r|n|} |D]{} |j|| | _ | j|| }|j|||j|j|t||| | t!|||| }y ) zxEnsures dest values off the surface work correctly when using the defaults for setcolor and unsetcolor. rrrrrrrrr N)r rr>rextendrCrMr!r rrrrrOrrVr[)rur<rr8r9rrrrrr!rrRrrs r!test_to_surface__dest_off_surfacez.MaskTypeTest.test_to_surface__dest_off_surfacesF"<<0#\\'2%% v..x4 U+ &6'"eVQK!fW> [!1!1!345! D;;##Dt#4D I15-;MN  ]+$( !!__W4_@  j'2  !4!4!6=#D*niP-*mY   rc dx\}}}tj|td}tjd}|j }tjd}|j ||j }tjd} |j | | | f| dfd| fg} | j t|j|||dd} t| } | jd d D]} tjj|| }|j}| r|n| }| D]}||_ d D]~}|j ||r| n| }||d <|jd i|}|j|||j!|j#|t%||||t'||||y)zkEnsures dest values off the surface work correctly when using setsurface and unsetsurface. rrrrrrNrrrrrr2)r r>rrrrrrCrMrrr!r rrrrOrrVr[)rur8r9rrrrrrrrrrrr!rrRrrrrs r>test_to_surface__dest_off_surface_with_setsurface_unsetsurfacezKMaskTypeTest.test_to_surface__dest_off_surface_with_setsurface_unsetsurfaces &% v..x4 U+ \\^ !<<0()||~ #\\&1,-&6'"eVQK!fW> [!1!1!345 $(  F| FG! D;;##Dt#4D I15-;MN $( !(5 $LL/2B,K*.K'!0!?;!?JMM*g6$$Z%8%8%:DA'j.)T1j-    rctjd}tjd}dx\}}tj|td}tjd}dD]}tjj ||}|j } | j} |r|n|} fdt|DD]} |j|| | _ | j| } d | _ |j|| }|j|||j|j|t!||| | t#|||| y ) ztEnsures area values on the mask work correctly when using the defaults for setcolor and unsetcolor. rrrrrrrc3FK|]}tD]}||f ywrFrrs rr7z=MaskTypeTest.test_to_surface__area_on_mask..%L1uU|L!ALLrr/rN)r rr>rr!r rMrr rrrLrrrOrrVr[)rur<rr9rrrrr!rrSrRrUrMrr8s @rtest_to_surface__area_on_maskz*MaskTypeTest.test_to_surface__area_on_masks8 "<<0#\\'2%% v..x4 U+ ! D;;##Dt#4D I!(I15-;MNMf L  ]+$' !(~~i8 '- $!__W9_E  j'2  !4!4!6=#D*nlS-*m\  rc dx\}}tj|td}tjd}|j }tjd}|j ||j }tjd}|j ||||tj d|d} t| } | jdd D]%} tjj|| } | j} | j }| r|n|}fd t|DD]}||_ | j|}d|_ d D]}|j ||r| n| }||d _ | j|d }d|_ | jdi|}|j!|||j#|j%|t'||||t)||||(y )zgEnsures area values on the mask work correctly when using setsurface and unsetsurface. rrrrrr/rrrrrrrc3FK|]}tD]}||f ywrFrrs rr7zZMaskTypeTest.test_to_surface__area_on_mask_with_setsurface_unsetsurface..(rrrNr2)r r>rrrrr'rrr!r rMr rrLrrrOrrVr[)rur9rrrrrrrrrrr!rrSrRrUrMrrrr8s @r:test_to_surface__area_on_mask_with_setsurface_unsetsurfacezGMaskTypeTest.test_to_surface__area_on_mask_with_setsurface_unsetsurfaces &% v..x4 U+ \\^ !<<0()||~ #\\&1,- $(KK-  F| FG! D;;##Dt#4D I!(I15-;MNMf L $' !(~~i8 '- $(5$LL/2B,K25K'/#,>>+f2E#FL+1L(!0!?;!?JMM*g6$$Z%8%8%:DA'j.,W1j-   rc tjd}tjd}dx\}}}tj|td}tjd}| | f| dfd| fg}|j t tj d||fdD]} tjj|| } | j} | j} | r|n|} |D]}|j||| _ | j| }d|_ | j|| }|j|||j!|j#|t%||| |t'||||y ) zuEnsures area values off the mask work correctly when using the defaults for setcolor and unsetcolor. rrrrrrr/rrrN)r rr>rrrCr'r!r rMrrrrLrrrOrrVr[)rur<rr8r9rrr positionsrr!rrSrRrUrMrs rtest_to_surface__area_off_maskz+MaskTypeTest.test_to_surface__area_off_mask=sp "<<0#\\'2%% v..x4 U+ fvg&% a&\B V[[%%IJK! D;;##Dt#4D I!(I15-;MN  ]+$' !(~~i8 '- $!__W9_E  j'2  !4!4!6=#D*nlS-*m\   rc dx\}}}tj|td}tjd}|j }tjd}|j ||j }tjd} |j | | | f| dfd| fg} | j ttjd||f|||tjd|d} t| } | jd d D]} tjj|| }|j}| r|n| }| D]}d D]}|j ||r| n| }||d _|j|d }d|_|j di|}|j#|||j%|j'|t)||||t+||||y )zhEnsures area values off the mask work correctly when using setsurface and unsetsurface. rrrrrrr/r rrrrNr2)r r>rrrrrrCr'rrr!r rMrrLrrrOrrVr[)rur8r9rrrrrrrrrrrr!rrRrUrrrMrs r;test_to_surface__area_off_mask_with_setsurface_unsetsurfacezHMaskTypeTest.test_to_surface__area_off_mask_with_setsurface_unsetsurfacebs &% v..x4 U+ \\^ !<<0()||~ #\\&1,-fvg&% a&\B V[[%%IJK $(KK-  F| FG! D;;##Dt#4D I15-;MN  (5$LL/2B,K25K'/#,>>+f2E#FL+1L(!0!?;!?JMM*g6$$Z%8%8%:DA'j.,W1j-   rcTd}d}tj|}tjjdd}|j |}|j ||t s%|jtj|||j|j|y)z2Ensures zero sized surfaces are handled correctly.rr/rrTrN) r r>r!r rrrrOrrr)rurrrr!rs r'test_to_surface__surface_with_zero_sizez4MaskTypeTest.test_to_surface__surface_with_zero_sizes..&{{T2__W-  j'*   S__Z8:L M ,,.5rcNd}t}d}tjd}d}tjj |d}tj d||}|j |}|j|tj ts%|jtj|||j|j|z|j|j||j|j|t!|||y ) z5Ensures zero sized setsurfaces are handled correctly.r rrr rTrr/rKNrr rr!r r>rrtrrOrrr|rrrrV) rurrrrRr9r!rrs r*test_to_surface__setsurface_with_zero_sizez7MaskTypeTest.test_to_surface__setsurface_with_zero_sizes g. {{ 5^^FM>J __ _;  j&..9   S__Z8:L M  ,,.>? //1>B ,,. :D*n=rcJd}t}d}tjd}d}tjj |}tj d||}|j |}|j|tj ts%|jtj|||j|j|z|j|j||j|j|t!|||y)z7Ensures zero sized unsetsurfaces are handled correctly.r rrrr/rPNr) rurrrrRr9r!rrs r,test_to_surface__unsetsurface_with_zero_sizez9MaskTypeTest.test_to_surface__unsetsurface_with_zero_sizes g. {{ *~~fm^L __,_?  j&..9   S__Z8:L M  ,,.>? //1>B ,,. :D*n=rc dD]~}dD]w}d|d|}tjj||}|j|tjj||j |j ||yy)z-Ensures masks can be created with zero sizes.)rr)rrr/rrz, fill=rN)r r!r rtrOr)rurrr]r!s rtest_zero_maskzMaskTypeTest.test_zero_masks0 =D% =dV74&1{{''4'8%%dFKK,<,   Y - D)T 2 3rcdD]C}tjj|}|j}|j ||Ey)z4Ensures get_size correctly handles zero sized masks.))rFr)r(r/N)r r!r rrOrurwr!rs rtest_zero_mask_get_sizez$MaskTypeTest.test_zero_mask_get_sizes@7 2M;;##M2D==?D   T= 1  2rcdD]Y}tjd|}tjj|}|j }|j ||[y)z4Ensures get_rect correctly handles zero sized masks.)r)rrr/r/N)r r'r!r rMrO)rurwrr!rBs rtest_zero_mask_get_rectz$MaskTypeTest.test_zero_mask_get_rectsP5 2M"KK >M;;##M2D==?D   T= 1  2rcdD]P}tjj|}|jt5|j d}dddRy#1swY]xYw)z2Ensures get_at correctly handles zero sized masks.))3r)rr`r/r/Nr)rurr!rs rtest_zero_mask_get_atz"MaskTypeTest.test_zero_mask_get_ats[. ,D;;##D)D"":. , F+ , , , , , AA cdD]P}tjj|}|jt5|j ddddRy#1swY]xYw)z2Ensures set_at correctly handles zero sized masks.))rr)rr/r/Nr r~s rtest_zero_mask_set_atz"MaskTypeTest.test_zero_mask_set_atsY. $D;;##D)D"":. $ F# $ $ $ $ $r+c d}tddD]s\}}d|d|}tjj|d}tjj|d}|j ||}|j ||uy) zwEnsures overlap correctly handles zero sized masks. Tests combinations of sized and zero sized masks. r/r)*size1=, size2=TrN)r:r r!r rr.)rursize1size2r]rxryrs rtest_zero_mask_overlapz#MaskTypeTest.test_zero_mask_overlap s +B3 0LE55'%1CKK$$U$6EKK$$U$6E--v6K   k3 / 0rcd}d}tddD]t\}}d|d|}tjj|d}tjj|d}|j ||}|j |||vy ) z|Ensures overlap_area correctly handles zero sized masks. Tests combinations of sized and zero sized masks. r/rrF4r1r2TrN)r:r r!r r_rO) rurrvr3r4r]rxryrHs rtest_zero_mask_overlap_areaz(MaskTypeTest.test_zero_mask_overlap_areas +B3 ALE55'%1CKK$$U$6EKK$$U$6E!..uf=M   ]NC @ Arcd}d}tddD]\}}d|d|}tjj|d}tjj|d}|j ||}|j |tjj||j |j|||j |j||y ) z|Ensures overlap_mask correctly handles zero sized masks. Tests combinations of sized and zero sized masks. r/rrrr1r2TrN) r:r r!r rZrtrOr^r) rurrvr3r4r]rxryrZs rtest_zero_mask_overlap_maskz(MaskTypeTest.test_zero_mask_overlap_mask*s +B3 BLE55'%1CKK$$U$6EKK$$U$6E --eV3 G   \224eS A Brcd}dD]U}tjj|}|j|j |j |d|Wy)z0Ensures fill correctly handles zero sized masks.rrrN)r r!r rrOr^)rurvrr!s rtest_zero_mask_fillz MaskTypeTest.test_zero_mask_fill=sS0 KD;;##D)D IIK   TZZ\>U4&> J  Krcd}|D]Q}tjj|}|j|j |j dSyNrr)r r!r rirOr^rur6rr!s rtest_zero_mask_clearz!MaskTypeTest.test_zero_mask_clearHsI, .D;;##D)D JJL   TZZ\1 - .rcd}|D]Q}tjj|}|j|j |j dSyr>)r r!r rrOr^r?s rtest_zero_mask_flipz MaskTypeTest.test_zero_mask_flipPsI, .D;;##D)D KKM   TZZ\1 - .rc d}|D]|}tjj|}|jd}|j |tjj|j |j d~y)Nrr r)r r!r rrtrOr)rur6rr!rys rtest_zero_mask_scalez!MaskTypeTest.test_zero_mask_scaleXsi, 7D;;##D)DJJv&E  ! !%)9)9 :   U^^-v 6  7rcd}tddD]\}}d|d|}tjj|d}tjj|d}|j }|j |||j |j |||j |j||y) ztEnsures draw correctly handles zero sized masks. Tests combinations of sized and zero sized masks. r/rrr1r2TrN)r:r r!r r^rjrOrrurr3r4r]rxryrvs rtest_zero_mask_drawz MaskTypeTest.test_zero_mask_drawbs +B3 ;LE55'%1CKK$$U$6EKK$$U$6E"[[]N JJuf %   U[[]NC @   U^^-uc : ;rcd}tddD]\}}d|d|}tjj|d}tjj|d}|j }|j |||j |j |||j |j||y) zuEnsures erase correctly handles zero sized masks. Tests combinations of sized and zero sized masks. r/rrr1r2TrN)r:r r!r r^rrOrrGs rtest_zero_mask_erasez!MaskTypeTest.test_zero_mask_erasets +B3 ;LE55'%1CKK$$U$6EKK$$U$6E"[[]N KKv &   U[[]NC @   U^^-uc : ;rcd}|D]C}tjj|d}|j|j dEy)NrTrr)r r!r rOr^r?s rtest_zero_mask_countz!MaskTypeTest.test_zero_mask_countsE, .D;;##Dt#4D   TZZ\1 - .rcd}|D]A}tjj|}|j|j dCy)Nrr/)r r!r rOrr?s rtest_zero_mask_centroidz$MaskTypeTest.test_zero_mask_centroids@, 6D;;##D)D   T]]_f 5 6rcd}|D]A}tjj|}|j|j dCy)Nrr)r r!r rOr r?s rtest_zero_mask_anglez!MaskTypeTest.test_zero_mask_angles@, 0D;;##D)D   TZZ\3 / 0rcg}dD]G}tjj|}|j}|j ||d|Iy)z3Ensures outline correctly handles zero sized masks.))rrr<r/rNr r!r rrgruexpected_pointsrr!r(s rtest_zero_mask_outlinez#MaskTypeTest.test_zero_mask_outlinesO. JD;;##D)D\\^F  E$. I  Jrcg}dD]H}tjj|}|jd}|j ||d|Jy)z_Ensures outline correctly handles zero sized masks when using the skip pixels argument.))rr)rrr/rarNrUrVs r test_zero_mask_outline__with_argz-MaskTypeTest.test_zero_mask_outline__with_argsT. JD;;##D)D\\"%F  E$. I  Jrc dD]}tjj|d}dD]}d|d|}tjj|d}td|d|dzdz td|d|dzdz f}|j |}|j |tjj||j ||||j|j||y ) zEnsures convolve correctly handles zero sized masks. Tests the different combinations of sized and zero sized masks. ))rrrrrFr/Tr)rQrrSr/sizes=rcrrN) r r!r r&r rtrrOr)rur3rxr4r]ryrwr!s rtest_zero_mask_convolvez$MaskTypeTest.test_zero_mask_convolves : FEKK$$U$6E< FugRw/ ((T(:58eAh.2358eAh.23! ~~e,%%dFKK,<,r/)r4r\r]r/rarcN)r r!r r rtrrOr) rur3rxr4ry output_sizer] output_maskr!s r)test_zero_mask_convolve__with_output_maskz6MaskTypeTest.test_zero_mask_convolve__with_output_masks : HEKK$$U$6E= H ((T(:#EHK"5'E7"[MBC"(++"2"2;"?K >>%=D))$ 0@0@#FMM$ S9$$T]]_k3GH H HrcVd}dD]}d|}tjj|}|j}|j |tjj||j |j ||j |j||y)z?Ensures connected_component correctly handles zero sized masks.r)r_)rPr/rN)r r!r rLrtrOrr^)rurvrr]r!cc_masks r"test_zero_mask_connected_componentz/MaskTypeTest.test_zero_mask_connected_components. CD$.C;;##D)D..0G  ! !'6;;+;+;S A   W--/ 6   W]]_nc B CrcdD]P}tjj|}|jt5|j d}dddRy#1swY]xYw)zdEnsures connected_component correctly handles zero sized masks when using an index argument.)rdrfr/r/N)r r!r r|rrL)rurr!rns r+test_zero_mask_connected_component__indexedz8MaskTypeTest.test_zero_mask_connected_component__indexeds_/ ;D;;##D)D"":. ;226: ; ; ; ; ;r+cg}dD]G}tjj|}|j}|j ||d|Iy)z@Ensures connected_components correctly handles zero sized masks.)r rr/rN)r r!r rhrg)ruexpected_cc_masksrr!cc_maskss r#test_zero_mask_connected_componentsz0MaskTypeTest.test_zero_mask_connected_componentssT. ND;;##D)D002H  +<dVn M  Nrcg}dD]G}tjj|}|j}|j ||d|Iy)z>Ensures get_bounding_rects correctly handles zero sized masks.))r r)rrr/rN)r r!r rrg)ruexpected_bounding_rectsrr!bounding_rectss r!test_zero_mask_get_bounding_rectsz.MaskTypeTest.test_zero_mask_get_bounding_rectssU"$. D;;##D)D!446N   75   rctjd}tjd}dD]}tj|td}|j |dD]z}tj j |d}|j||}|j|||j|j|d |vsnt||||y ) zCEnsures to_surface correctly handles zero sized masks and surfaces.rr)r)rr)rrr/r))rr)rrr/TrrrN) r rr>rrr!r rrrOrrV)ru mask_color surf_colorrrr9r!rs rtest_zero_mask_to_surfacez&MaskTypeTest.test_zero_mask_to_surface s\\&) \\%( 9 FInnY"=G LL $5 F {{'' '=!__Wz_J  j'2  !4!4!6 BI%'j*E F Frc"tjd}dD]u}tjj|d}|j |}|j |tj |j|j|wy)z|Ensures to_surface correctly handles zero sized masks and surfaces when it has to create a default surface. r)rr/r/TrrN) r rr!r rrtr>rOr)rur{r9r!rs r)test_zero_mask_to_surface__create_surfacez6MaskTypeTest.test_zero_mask_to_surface__create_surface!sw\\&) 1 ?I;;##ID#9D*=J  ! !*fnn =   Z002I >  ?rN)F)__name__ __module__ __qualname__r#unittestskipIfrrzrrrrrrrrrrrrrrrrrrrrr r rr&r+r2r6r;r@rDrIrNrPrSrVrXr]rgrkrurxrzr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrr$r(r-r5r9r=r@rDrPrSrUrYr^rardrqrurxrrrrrrrrrrrrrexpectedFailurerrrrrrrrrrrrr rrrrr r%r'r)r0r2r>rCrIrNrRrVrYr`rdrfrlrnrprrrtrwryr{rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrrrrr!r%r'r*r.r5r8r:r<r@rBrErHrKrMrOrQrXrZrbrkrorqruryr}rr2rrreres  NX__W12:3: ) BX__W12F3F( BC&;$ ; ;*9 2X__W125:35:n .)X__W12 .3 ..2 M!">$>$>$!"!GF@.">H<8@<X__W12G3G:X__W12=3=7X__W12!G3!GFX__W12@3@:@>X__W12!G3!GFD>"GHX__W127372 @DX__W12!D3!DF @DX__W12&G3&GPC=D& 9X__W123$ 9&X__W12838> 9 9X__W123*X__W12S3S6 !G@@B$=L@8X__W12(G3(GT,&G@@B$=L@8X__W12(G3(GT-'9 X__W123< 9 96 9>">&>">& 6>0 0 9'*M( 0 '2 DX__W12 3 ",,*XB$B&! F$ L=. F):V5(5*05dX__W12w3wrX__W12B3B(F2B0B6B4B0 >B4 >B@R.BB>R0B>BB>(B>%*N&+BZB*B* 3 7B" + B BVV@??$QQ0BB*RVhUVn# J?$BB0J@ D MD"HV4B.B0B0J@ D MD"HV4B0B0V6B2B2V8 JD"HV8 JD"HV8 JD"HX__W12B3BH!FX__W12"3"H33j))VB4B$F>CBR@H:E:G>V,:1f@3jX__W123@66pX__W12!3!FX__W12434l 6>(>( = 322,$0 A"B& K..7;$;$.60 J JF,H, C; N F( ?rrec"eZdZdZfdZxZS)SubMaskz4Subclass of the Mask class to help test subclassing.c2t||i|d|_y)NT)super__init__test_attribute)rurr __class__s rrzSubMask.__init__3s $)&)"r)rrr__doc__r __classcell__rs@rrr0s>##rrc"eZdZdZfdZxZS) SubMaskCopy;Subclass of the Mask class to help test copying subclasses.cFt|}|j|_|SrF)rrrrurrs rrzSubMaskCopy.copy;s"GLN #'#6#6  rrrrrrrrs@rrr8Errc"eZdZdZfdZxZS)SubMaskDunderCopyrcFt|}|j|_|SrF)r__copy__rrs rrzSubMaskDunderCopy.__copy__Ds$G$& #'#6#6  r)rrrrrrrs@rrrArrrc"eZdZdZfdZxZS)SubMaskCopyAndDunderCopyrc t|SrF)rr)rurs rrzSubMaskCopyAndDunderCopy.copyMsw|~rrrs@rrrJsErrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZy)MaskSubclassTestzTest subclassed Masks.ctdd}|j|tjj|j|t|j |j y)z)Ensures the Mask class can be subclassed.r=TrN)rrtr r!r r|rrs rtest_subclass_maskz#MaskSubclassTest.test_subclass_maskTsJvD) dFKK$4$45 dG, ++,rcftdd}|jtj|fD]|}|j|tjj |j|t|j ||t||||jt|d~y)z(Ensures copy works for subclassed Masks.rr TrrN) rrrtr r!r rrc assertFalsehasattrrur!rs rtest_subclass_copyz#MaskSubclassTest.test_subclass_copy\swT*))+tyy7 CI  ! !)V[[-=-= >  ! !)W 5   Y - D)T 2   WY0@A B  Crctdd}t|jtj|fD]\}}|j|tj j |j|t|j||t|||d|k(r|jt|d|j|jy)z8Ensures copy works for subclassed Masks overriding copy.rTrrrN) rrrrtr r!r rrcrrr|r)rur!rrs r!test_subclass_copy__override_copyz2MaskSubclassTest.test_subclass_copy__override_copyis7.&tyy{DIIdO&DE :LAy  ! !)V[[-=-= >  ! !)[ 9   Y - D)T 2Av  4D!EF 8 89 :rcftdd}|jtj|fD]|}|j|tjj |j|t|j ||t||||j|j~y)z  ! !)-> ?   Y - D)T 2 OOI44 5 6rcftdd}|jtj|fD]|}|j|tjj |j|t|j ||t||||j|j~y)zAEnsures copy works for subclassed Masks overriding copy/__copy__.rTrN) rrrtr r!r rrcr|rrs r.test_subclass_copy__override_both_copy_methodsz?MaskSubclassTest.test_subclass_copy__override_both_copy_methodss'd;))+tyy7 6I  ! !)V[[-=-= >  ! !)-E F   Y - D)T 2 OOI44 5  6rcbd}t|}|j}|j||y)z,Ensures get_size works for subclassed Masks.rDN)rrrOr$s rtest_subclass_get_sizez'MaskSubclassTest.test_subclass_get_sizes, }%}} }-rctjdd}t|jd}|j }|j ||y)z,Ensures get_rect works for subclassed Masks.r/)rrTrN)r r'rrrMrO)rurr!rBs rtest_subclass_mask_get_rectz,MaskSubclassTest.test_subclass_mask_get_rects? FH5 }))5}} }-rchd}tdd}|jd}|j||y)z*Ensures get_at works for subclassed Masks.rrTrr/N)rrrO)rurr!bits rtest_subclass_get_atz%MaskSubclassTest.test_subclass_get_ats1 vD)kk&! l+rcd}d}d}tdd}|j||j|j|||j|j |y)z*Ensures set_at works for subclassed Masks.rr/FrrN)rrrOrr^)rurrvrUr!s rtest_subclass_set_atz%MaskSubclassTest.test_subclass_set_atsX E/ C S)<8 ~6rc$d}d}tjjd|t|df}tjjd|t|df}|D]-}|D]&}|j |d}|j ||(/y)z+Ensures overlap works for subclassed Masks.r/rDTrN)r r!r rrrO)rurr9masks arg_masksr!arg_maskrs rtest_subclass_overlapz&MaskSubclassTest.test_subclass_overlaps  !!t)!VW KK  $Y  7 It $  VW KK  $Y  7 It $  @D% @ $ 1 1(F C   ? @ @rcd}|d|dz}tjjd|t|df}tjjd|t|df}|D]}|D]}|j |d}|j |tjj|j |t|j|j||j|j|y)z0Ensures overlap_mask works for subclassed Masks.rrrTrr/N) r r!r rrZrtassertNotIsInstancerOr^r)rurwrvrrr!rrZs rtest_subclass_overlap_maskz+MaskSubclassTest.test_subclass_overlap_masks &q)M!,<< KK  $]  ; M4 (  KK  $]  ; M4 (  ID% I#006B %%lFKK4D4DE((w?  !3!3!5~F  !6!6!8-H  I Ircd}|d|dz}td|}|j|j|j|y)z(Ensures fill works for subclassed Masks.rrrFrN)rrrOr^rur9rvr!s rtest_subclass_fillz#MaskSubclassTest.test_subclass_fillsD "1 ! 4E 2  ~6rcd}d}t|d}|j|j|j|y)z)Ensures clear works for subclassed Masks.)rrrTN)rrirOr^rs rtest_subclass_clearz$MaskSubclassTest.test_subclass_clears7 y$'  ~6rcd}|d|dz}td|}|j|j|j|y)z*Ensures invert works for subclassed Masks.rrrrFrN)rrrOr^rs rtest_subclass_invertz%MaskSubclassTest.test_subclass_invertsD "1 ! 4E 2  ~6rcd}td}|j|}|j|tjj |j |t|j|j|y)z)Ensures scale works for subclassed Masks.r~rN) rrrtr r!r rrOr)rurwr! scaled_masks rtest_subclass_scalez$MaskSubclassTest.test_subclass_scalesa vjj/  k6;;+;+;<   g6 --/?rchd}|d|dz}tjjd|t|df}tjj|t|fD]K}|D]D}|j |j |d|j |j|FMy)z(Ensures draw works for subclassed Masks.rrrTrr/N)r r!r rrirjrOr^)rur9rvrr!rs rtest_subclass_drawz#MaskSubclassTest.test_subclass_draws "1 ! 4 KK  $Y  7 It $ [[%%i0')2DE ?D% ?  (F+  ~>  ? ?rc\d}d}tjj|dt|df}tjj|dt|df}|D]K}|D]D}|j |j |d|j |j|FMy)z)Ensures erase works for subclassed Masks.rrTr/N)r r!r rrrrOr^)rur9rvrrr!rs rtest_subclass_erasez$MaskSubclassTest.test_subclass_erase/s !!)T2GIt4LM[[%%i6 48PQ  ?D% ?  8V,  ~>  ? ?rcd}|d|dzdz }td|}|jdd|j}|j||y)z)Ensures count works for subclassed Masks.r~rrTrrgN)rrr^rO)rur9rvr!r^s rtest_subclass_countz$MaskSubclassTest.test_subclass_count?sS "1 ! 4q8Dy1 FA  /rcfd}d}td}|j}|j||y)z,Ensures centroid works for subclassed Masks.r/rN)rrrO)rurr9r!rs rtest_subclass_centroidz'MaskSubclassTest.test_subclass_centroidJs2" v==? #45rcdd}td}|j}|j||y)z)Ensures angle works for subclassed Masks.rrrsN)rr r)rur r!r s rtest_subclass_anglez$MaskSubclassTest.test_subclass_angleTs,F#  un5rcbg}td}|j}|j||y)z+Ensures outline works for subclassed Masks.rN)rrrg)ruexpected_outliner!rs rtest_subclass_outlinez&MaskSubclassTest.test_subclass_outline]s-v,,. W&67rc d\}}||f}d}td|dzdz td|dzdz f}tjj|t |f}tjj|t |f}tjj|t |fD]]}|D]T} |j | } |j | tjj|j| t|j| j||j| j||D]} |j | | } |j | tjj|j| j||j| j|t| tr|j | t|j| tW`y)z,Ensures convolve works for subclassed Masks.r4rr rN) r&r r!r rr rtrrOr^r isinstance) rur8r9r9rvrwr output_masksr!rr'rjs rtest_subclass_convolvez'MaskSubclassTest.test_subclass_convolvefs vFO Q A .AvzA~0FG [[%%i0')2DE  ((3WY5GH [[%%i0')2DE ID% I $ h 7 %%mV[[5E5EF((@  !4!4!6G  !7!7!9=I$0 IK$(MM(K$HM))-9I9IJ$$]%8%8%:NK$$]%;%;%=yI!+w7--mWE00H I I IrcBd}d}t|}|j}|j|tjj |j |t|j|j||j|j|y)z7Ensures connected_component works for subclassed Masks.rrN) rrLrtr r!r rrOr^r)rurvrwr!rns r!test_subclass_connected_componentz2MaskSubclassTest.test_subclass_connected_components} }%**, gv{{'7'78   '2 .9 ))+];rcbg}td}|j}|j||y)z8Ensures connected_components works for subclassed Masks.rN)rrhrg)ru expected_ccsr!ccss r"test_subclass_connected_componentsz3MaskSubclassTest.test_subclass_connected_componentss. v'') S,/rcbg}td}|j}|j||y)z6Ensures get_bounding_rects works for subclassed Masks.rN)rrrg)rurwr!rxs r test_subclass_get_bounding_rectsz1MaskSubclassTest.test_subclass_get_bounding_rectss0"$v002 ^-DErcntjd}d}t|d}tj|td}|j tjd|j ||}|j|||j|j|t|||y) z.Ensures to_surface works for subclassed Masks.rr=TrrrrN) r rrr>rrrrrOrrVrs rtest_subclass_to_surfacez)MaskSubclassTest.test_subclass_to_surfacesf-t$'..x4 V\\%()__W~_F  j'* ,,.5D*n=rN)rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr2rrrrQs - C:" 6 6.., 7<"@"I.777 @?$? 0668IB <0F >rrzpypy has lots of mask failurescHeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) MaskModuleTestc fd}gd}d}|d|dz}d}dD]}tj|t|}t|D]}||d<|j ||dkr|j d d}d d|dz ||dzd dh} | D]} d |d |d| } || kDr|} n|} tj j|| } |j| tj j| |j| j|| |j| j| | y)aEnsures from_surface creates a mask with the correct bits set. This test checks the masks created by the from_surface function using 16 and 32 bit surfaces. Each alpha value (0-255) is tested against several different threshold values. Note: On 16 bit surface the requested alpha value can differ from what is actually set. This test uses the value read from the surface. )7rr)rqrwrrrrrr/rjrdepth=z, alpha=z , threshold=)rr N) r r>rr rrr!r rtr rOrr^)ruthreshold_countrrw all_set_countnone_set_countrralphathreshold_test_valuesr r]rvr!s rtest_from_surfacez MaskModuleTest.test_from_surfaces^)  %a(=+;;  HEnn]HeDG/ H#( a  ]+2:$NN6215E*,Q 5%!)SRU(V%!6HI"5'% YKPCy()6)7!;;33 '94D))$ 0@0@#F$$T]]_mSI$$TZZ\>3GH H Hrcd}d}gd}|df}tj|d}tj|td}|j t |D]}||d<|j |d f||jt |D]}d |} |j |d fd |j} tjj||} |j| tjj| |j| j|| |j| j| | |j| j||| | y ) aEnsures from_surface creates a mask with the correct bits set when pixels have different alpha values (32 bits surfaces). This test checks the masks created by the from_surface function using a 32 bit surface. The surface is created with each pixel having a different alpha value (0-255). This surface is tested over a range of threshold values (0-255). r/r)rarr-rrTrrrr threshold=N)r r r>rrNr rrPr^r!r rtrOrr_) rurrrrwr\rr4r r]rvr!s r)test_from_surface__different_alphas_32bitz8MaskModuleTest.test_from_surface__different_alphas_32bitsN' (!,  M= .."=  ' 2A M!  NNAq6= 1 2 / Iyk*C  )Q 3*002N;;++GY?D  ! !$ (8(8# >   T]]_mS A   TZZ\>3 ?   !!-8.#  rc&d}gd}|df}tj|td}|jt |D]}||d<|j |df||j t}t}t |D]N}|j|dfd} | |vr|g|| <n|| j||| ks>|j|Pg} d} tj|} tj|d } |jD]}|D]}||vr| j||dz| f|d dz}t |D]}| j |dfdtj|d } | j| | | j||| f| D]\}}}|j}t ||D]}d |}tj j#||}|j%|tj j||j'|j)|||j'|j|||j'|j+|| ||y ) aEnsures from_surface creates a mask with the correct bits set when pixels have different alpha values (16 bit surfaces). This test checks the masks created by the from_surface function using a 16 bit surface. Each pixel of the surface is set with a different alpha value (0-255), but since this is a 16 bit surface the requested alpha value can differ from what is actually set. The resulting surface will have groups of alpha values which complicates the test as the alpha groups will all be set/unset at a given threshold. The setup calculates these groups and an expected mask for each. This test data is then used to test each alpha grouping over a range of threshold values. r)nxrrrrrr/TrrjrN)r r>rrNr rrPrsetrrJaddr rrr^r!r rtrOrr_)rurrrwrr4alpha_thresholdsspecial_thresholdsr rrr erase_maskexp_mask thresholds to_thresholdthresfrom_thresholdr\rvr]r!s r)test_from_surface__different_alphas_16bitz8MaskModuleTest.test_from_surface__different_alphas_16bit s* (!, .."=  ' 2A M!  NNAq6= 1 2 &= U/ 2INNIq>215E,,+4+ ' '..y95 "&&y1 2 [[/ ;;}48*113 J'   22$$iQ%IJ#-b>A#5L"'|!49"))5!*a89 &{{=tDHNN:v6$$ix%HI  &<@  "9+.{{//C%%dFKK,<, set_colorkeyrrr!r rtr rOrr^) ru colorkeysrwrvrr]rcolorkeyr!s r-test_from_surface__with_colorkey_mask_clearedzrr r!r rtr rOrr^) rur rrwrvrr]rr r!s r,test_from_surface__with_colorkey_mask_filledz;MaskModuleTest.test_from_surface__with_colorkey_mask_filled|sL &  &q)M!,<<$ DE5'"Cnn]Au=G LL '% D$$X.{{//8%%dFKK,<,.alternate..r;rr)func set_value unset_valuer8r9setbitrUs ` r alternatezOMaskModuleTest.test_from_surface__with_colorkey_mask_pattern..alternates5FLe L $Sv);?# $r)rrar)r`rTr^)rqr rrr/rrN) r r!r rr^r>r r rtrOrr_) rurrr rwr\rvrrr]rr!s r-test_from_surface__with_colorkey_mask_patternz   T]]_mS A   TZZ\>3 ?   !!-8.#  rcgd}|D]}tjjdd|}|jddtjj |dd}|j }|j|jd|j|j tjdg|D]}tjjdd|}tjjdd|}|jd |jd |jd d tjj |d d| }|j|tjj|j|jd|j|j tjd gy)z*Does mask.from_threshold() work correctly?rrr)r^r^rrr`rrrrrrr`rrrarararirrrr r r#r#rarar)rcolorr othersurfacerN) r rr>rr!rrrOr^r'rtr )rur4rsurfr!rsurf2s rtest_from_thresholdz"MaskModuleTest.test_from_thresholds  YA>>))(Aq9D IIn&6 7;;--)+<D++-E   TZZ\3 /   T446EU9V8W X Y YA>>))(Aq9DNN**8Q:E IIo & JJ ' JJ(8 9;;--"+" .D  ! !$ (8(8 9   TZZ\3 /   T446EU9V8W X YrcdD]}tjjtj|}|j |tjj d||j |j|y)z?Ensures from_surface can create masks from zero sized surfaces.rrN)r r!r r>rtMaskTyperOrr~s rtest_zero_size_from_surfacez*MaskModuleTest.test_zero_size_from_surfacesf0 4D;;++FNN4,@AD  ! !$ (<(<dVn M   T]]_d 3  4rcvgd}d}|D]}|D]}tjj|d|}|jddtjj |dd}|j |jd|j}|j |g|D]}tjj|d|}tjj|d|}|jd|jd |jdd tjj |d d|}|j|tjj|j |jd|j}|j |g y) Nrrrrrrrrrr!r) r rr>rr!rrOr^rrtr ) rur4r6rrr$r!rr%s rtest_zero_size_from_thresholdz,MaskModuleTest.test_zero_size_from_thresholdsz , ,D ,~~--dAq9 .*:;{{11-/@  q1//1  + , ,~~--dAq9..tQ: /* ?+ ?,<={{11,(95%%dFKK,<,<=  q1//1  + , ,rc~d}d}d}tj|}|D]}|j|dt|}d|jdz}|D]1}|\}} ||z} |j || | f||zz dzdd|d3|D]1}|\}} ||z} |j || | f||zz dzdd|d 3y) N)ir)rf)rra)rg)r/)rra)r-rerr~z the pixel at z is not set to 1rz is not set to 0)r r r memoryviewstridesrO) rur pixels_set pixels_unsetr!pointviewintwidthrrcols rtest_buffer_interfacez$MaskModuleTest.test_buffer_interface s 3 5 {{4  "E KKq ! "$t||A& EDAqx-C   c1f!h,/14w&67  " EDAqx-C   c1f!h,/14w&67  rN) rrrrrrr rrr&r)r+r6r2rrrrs?.H`$LRhD6D4&P YD4,Brr__main__)r)rF)# collectionsrrplatformr rrr pygame.locals pygame.mathrpython_implementationrrr#r-r:r<rCrVr[rcTestCaserer!r rrrrrrrrmainr2rrr@s#    2H224 4   IFN*( I[V?8$$[V?|l#fkk#'0a>x((a>H :;oX&&o<od  zHMMOr