rbiddlmZddlZddlZddlmZmZmZm Z m Z m Z ddl Z ddl m Z mZmZddl mZddlZddlmZ ddlmZdZn #e$rd ZYnwxYweje d ZGd d ZGddZejjejdGddZ GddZ!dS))sqrtN)GeometryCollection LineString MultiPolygonPointPolygonbox) GeoDataFrame GeoSeries read_file)_compat)assert_array_equal) coo_arrayTFzscipy not installedreasonc>eZdZdZdZdZdZdZdZdZ dZ d S) TestSeriesSindexctgd}tgd}td||gid}|jrJ|j|jsJd|jj_|jrJ|j|jsJt||g}|jrJ|j|jsJd|j_|jrJ|j|jsJdS)zTest the has_sindex method.rrrrrrrrrgeomgeometryN)rr has_sindexsindexrvalues_sindexr )selft1t2dss k/var/www/html/mdtn/previsions/meteo_cartes/venv/lib/python3.11/site-packages/geopandas/tests/test_sindex.pytest_has_sindexz TestSeriesSindex.test_has_sindex s --- . . --- . . &2r(+f = = =< |$( !< | r2h  < |< |cvtt}|jrJt|jdksJdS)z7Tests creating a spatial index from an empty GeoSeries.dtyperN)r objectr lenr#r's r(test_empty_geoseriesz%TestSeriesSindex.test_empty_geoseries7s? F # # #818}}!!!!!!r*cVttddg}|jjdksJ|jd}t t |dksJ|jd}t t |dksJdS)Nrrr4rr)r5r4r4)r rr size intersectionr/list)r#r'hitss r( test_pointzTestSeriesSindex.test_point=s uQ{{m $ $x}!!!!x$$^444::!####x$$%5664::!######r*cttg}|jrJt|jdksJdS)z9Tests that a single empty Point results in an empty tree.rN)r rr r/r0s r(test_empty_pointz!TestSeriesSindex.test_empty_pointEsB uwwi 818}}!!!!!!r*ctgd}tgd}tgd}t|||g}|jjdksJdS)Nrr)rrrr)rr r r6)r#r$r%sqr's r( test_polygonszTestSeriesSindex.test_polygonsKsl --- . . --- . . 555 6 6 r2rl # #x}!!!!!!r*cttddg}|jjJ|jjdksJ|jjJdS)Nrr)r rr!r"r r6r0s r(test_lazy_buildz TestSeriesSindex.test_lazy_buildRsV uQ{{m $ $x'''x}!!!!x+++++r*cttddg}|j}tdd|jd<|j|usJdS)Nrr rr iloc)r#r'original_indexs r(test_rebuild_on_item_changez,TestSeriesSindex.test_rebuild_on_item_changeXsL uQ{{m $ $!QKKq x~------r*c ttddtddg}|j}|jdd}|j|usJ|jdd}|j|usJ|jddd}|j|usJdS)Nrrr4rD)r#r'rFsliceds r(test_rebuild_on_slicez&TestSeriesSindex.test_rebuild_on_slice^s uQ{{E!QKK0 1 1}N2222}...."}N222222r*N) __name__ __module__ __qualname__r)r1r:r<r@rBrGrJr*r(rrs.""" $$$""" """,,, ... 3 3 3 3 3r*rcDeZdZdZdZdZdZdZdZdZ dZ d Z d S) TestFrameSindexc tdtdddttdtdDd}t|d|_dS)Nrc4g|]\}}t||SrNr.0xys r( z0TestFrameSindex.setup_method..q$EEETQU1a[[EEEr*)ABrrr)rangezipr dfr#datas r( setup_methodzTestFrameSindex.setup_methodmscqr1EESq588-D-DEEE   tf555r*cd|j_|jjjdksJt |jjd}t |dksJ|ddksJdS)Nz epsg:4326rR)@rerfrr>)r`crsr r6r8r7r/)r#r9s r( test_sindexzTestFrameSindex.test_sindexusk! w~"a''''DGN//0@AABB4yyA~~~~Aw!||||||r*c|jjjjJ|jjjdksJ|jjjjJdS)NrR)r`rr!r"r r6r#s r(rBzTestFrameSindex.test_lazy_build|sNw&.666w~"a''''w&.:::::r*c |jjJ|jj}|jdtt ddt ddDd|jj|usJdS)Nc4g|]\}}t||SrNrUrVs r(rZzGTestFrameSindex.test_sindex_rebuild_on_set_geometry..s$ E E ETQU1a[[ E E Er*rR T)inplace)r`r set_geometryr_r^)r#rFs r(#test_sindex_rebuild_on_set_geometryz3TestFrameSindex.test_sindex_rebuild_on_set_geometrysw~)))  E ESq"uQ||%D%D E E Et    w~^333333r*c|jj}|jjdd}|j|usJ|jj}|jjdd}|j|usJ|jjddd}|j|usJdS)Nrr4)r`r rE)r#rFrIs r(test_rebuild_on_row_slicez)TestFrameSindex.test_rebuild_on_row_slicesbqb!}N2222aaa}....ddd#}N222222r*c||jj}|jd}|j|usJ|jj}|j|usJdS)z?Selecting a single column should not rebuild the spatial index.rN)r`r r)r#rF geometry_cols r($test_rebuild_on_single_col_selectionz4TestFrameSindex.test_rebuild_on_single_col_selectionsPwv "n4444w' "n444444r*c|jj}|jddg}tjs |j|usJn |j|usJ|jddg}tjs |j|usJdS|j|usJdS)z2Selecting a subset of columns preserves the index.rr\N)r`r compat PANDAS_GE_30)r#rFsubset1subset2s r(&test_rebuild_on_multiple_col_selectionz6TestFrameSindex.test_rebuild_on_multiple_col_selections'63-(" 4>77777>^3333'3-(" 4>777777>^333333r*c|j}|j}|ddd|jrJ|j}||usJ|jgdksJdS)Nr\FT) ascendingro)rfr>rgrr)r`copyr sort_valuesrindextolistr#gdf old_sindex new_sindexs r(test_rebuild_on_update_inplacez.TestFrameSindex.test_rebuild_on_update_inplacesgllnnZ  ud;;;>!!!Z ++++y!!___444444r*c|j}|j}|ddid|jsJ|j}||usJdS)Nr\AAT)columnsro)r`rr renamerrs r(test_update_inplace_no_rebuildz.TestFrameSindex.test_update_inplace_no_rebuilds^gllnnZ  C; 555~Z Z''''''r*N) rKrLrMrcrirBrqrsrvr|rrrNr*r(rPrPls666;;; 444 3 3 3555444" 5 5 5(((((r*rP_setup_class_nybb_filenameceZdZdZdZdS)TestJoinSindexc8t|j|_dS)N)r nybb_filenameborosrks r(rczTestJoinSindex.setup_methodst122 r*cjj}|d}fd|D}|ddgksJjjddkj}|d}fd|D}|dgksJjjddkj}|d}fd|Df}|dgksJd t d ksJjjd ksJj}|d}fd |D}|ddgksJdS) N)g.AgHzb Ac@g|]}jj|dSBoroName)rrE)rWhitr#s r(rZz1TestJoinSindex.test_merge_geo..s'@@@Ctzs#J/@@@r*BronxQueensBoroCoder>c6g|]}j|dSrrE)rWrfirsts r(rZz1TestJoinSindex.test_merge_geo..s$;;;suz#z*;;;r*c6g|]}j|dSrr)rWrseconds r(rZz1TestJoinSindex.test_merge_geo..s%=== C ,===r*outer)howrRc6g|]}j|dSrr)rWrmergeds r(rZz1TestJoinSindex.test_merge_geo..s%<<<v{3 +<<d?Zejejd@/dAZejdBd gfegfegfegfegfgdCZdDZejdEdFgdGgdHgfdIgdGgdHgffdJZdKZejddej de dfejdd dggggfd dgdgdggfd dgddgddggfddggggfddgdgdggfddgddgddggfdddgdgdggfddLdggdMgdNgfddOggggfddPggggfddQgdgd!ggfd"dggggfd"dRgdgdggfd"dSgdgdggfd"dTgdgdggfd"dUgddgddggfd"dVgdgd!ggfd&dWgdgdggfd&dXggggfd(dYgddgddggfd(dZggggfd(dgddgddggfd+dggggfd+dRgdgdggfd+dSgdgdggfd+dTgdgdggfd+dUgddgddggfd+dVggggffd[Zejd\eggggfed ggggfd ggggfd e d d ddd gdgdggfgd]Z d^Z!d_Z"d`Z#ejdd dgggfddgggfd"dLdgdggffdaZ$ejdEdFgdGgdHgfdIgdGgdHgffdbZ%ejdcdFdIgejddddgdgdggfddgdgdggfgdeZ&ejdcdFdIgejdddfdgddgddggfdfdggddgddggfgdhZ'ejdcdFdIgejddd gggfd ggggfgdiZ(ejdjdFdIgejdkdFd gdMgdlgedededmgffdId ddgdd!ggededmgffdFdddgddggededgffdIddgdggedgffgdnZ)ejdjdFdIgejdodId dIgdpgdpgd!dgzffdId dFgdpgdqgd!ed%gzffdFd dIgdpgdpgd!dgzffdFd dFgdrgdsgdted%gzffdIdudFgdvgdwgdxdgzffdFdudFgdygdzgd{dgzffgd|Z*d}Z+d~Z,dZ-dZ.ejdgddZ/d S)TestShapelyInterfacecddttdtdDtddddgzi}t|d|_t |d|_dS)Nrc4g|]\}}t||SrNrUrVs r(rZz5TestShapelyInterface.setup_method..r[r*rRrnr)r_r^r r r`r/ expected_sizeras r(rcz!TestShapelyInterface.setup_methods| EESq588-D-DEEE2r2r""#$ tf555 f..r*ztest_geom, expectedr4r4r)rr?rr)rrrrrrc~t|jj|}t ||dS)z2Tests the `intersection` method with valid inputs.N)r8r`r r7r)r# test_geomexpectedrs r(test_intersection_bounds_tuplez3TestShapelyInterface.test_intersection_bounds_tuples947>..y99::3)))))r*r)r4r4rrNctjt5|jj|ddddS#1swxYwYdS)z4Tests the `intersection` method with invalid inputs.N)pytestraises TypeErrorr`r r7r#rs r(&test_intersection_invalid_bounds_tuplez;TestShapelyInterface.test_intersection_invalid_bounds_tuples]9 % % 3 3 GN ' ' 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 AA A  output_formatindicessparse)marksdensezpredicate, test_geom, expectedr4rrr intersectswithin?Q??rn rRcontainsjt??rgtouchesrcoversMbP?Gz?contains_properlyct|jj||}t|||dkrt jt |jt}d||<|jj|||}|dkr|}t||dSdS)z@Tests the `query` method with valid inputs and valid predicates. predicaterr,TrrrN r`r queryrnpzerosr/booltodense)r#rrrrrrs r( test_queryzTestShapelyInterface.test_querysNgn""9 "BB3))) I % %HS\\666E"E(O'.&&Ym'C((kkmm sE * * * * * & %r*ctjt5|jjdddddS#1swxYwYdS)z/Tests the `query` method with invalid geometry. notavalidgeomNrrrr`r rrks r(test_query_invalid_geometryz0TestShapelyInterface.test_query_invalid_geometrys ]9 % % 2 2 GN  1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2rzRequires GEOS 3.10rzdistance, test_geom, expectedg"@g#@g& .>r)r)rgr)rgrg)rrrrg)rrrrrrr)rrrgrrrgr>)rrrrrrr)rrrrrgr>rfcj|jj|d|}t||dS)zHTests the `query` method with predicates that require keyword arguments.dwithinrdistanceNr`r rr)r#rrrrs r(test_query_dwithinz'TestShapelyInterface.test_query_dwithins9pgn""9 H"UU3)))))r*ctjtd5|jjt dddddddS#1swxYwYdS)`Tests the `query` method with keyword arguments that are invalid for certain predicates.z8'distance' parameter is required for 'dwithin' predicatematchrrrNrr ValueErrorr`r rrrks r(test_dwithin_no_distancez-TestShapelyInterface.test_dwithin_no_distances] X    C C GN q! B B B C C C C C C C C C C C C C C C C C Cs0AA Ar) Nrr covered_byrcrossesroverlapsrrcd}tjt|5|jjt dd|dddddS#1swxYwYdS)rzD'distance' parameter is only supported in combination with 'dwithin'rrrNr)r#rmsgs r(test_query_distance_invalidz0TestShapelyInterface.test_query_distance_invalids$U ]:S 1 1 1 O O GN q! A N N N O O O O O O O O O O O O O O O O O Os1AA #A z0Test for 'dwithin'-incompatible versions of GEOSctjtd5|jjt ddddddddS#1swxYwYdS)zTests whether a ValueError is raised when trying to use dwithin with incompatible versions of shapely or pyGEOS z-predicate = 'dwithin' requires GEOS >= 3.10.0rrrrNrrks r(test_dwithin_requirementsz.TestShapelyInterface.test_dwithin_requirementss] M    O O GN q! A N N N O O O O O O O O O O O O O O O O O Os1AA!Aztest_geom, expected_valuecd|jj|}t||dS)z-Tests the `query` method with empty geometry.Nr)r#rexpected_valuers r(test_query_empty_geometryz.TestShapelyInterface.test_query_empty_geometrys1gn""9--3/////r*ctdddd}tjt5|jj|dddddS#1swxYwYdS)1Tests the `query` method with invalid predicates.r4rtestrN)r rrrr`r rrs r(test_query_invalid_predicatez1TestShapelyInterface.test_query_invalid_predicatesBd++ ]: & & > > GN f = = = > > > > > > > > > > > > > > > > > >s"AA"Azsort, expectedT)rrrrrrgFc ttgdg}ttgdtgdtgdg}gd}|jd}|j||}t |t |ksJ t ||d S#t$r8}|dur-tj d|d | d |d }~wwxYw) zYCheck that results from `query` don't depend on the order of geometries. r)r>rr>r>rr>)r4rrr)r4r>r)rRr>)rRrR)r>rRrrsortFkrtree results are known to be unordered, see https://github.com/geopandas/geopandas/issues/1337 Expected:  Got:  N) r rr!r rsortedrAssertionErrorrxfailr)r#rr test_polys tree_polystest_geores r(test_query_sortingz'TestShapelyInterface.test_query_sortingsM(H(H(H I IJKK 88899:::;;88899   99$Q'%%hT%::c{{fX......  sH - - - - -   u}} /#+//"jjll/// G s/C D 3C>>Dctjtd5tdddd}|jj|dddddS#1swxYwYdS)Nz#Invalid output_format: 'dataarray'.rr4r dataarray)r)rrrr r`r rrs r(test_unsupported_outputz,TestShapelyInterface.test_unsupported_output0s ]:-R S S S G GBD$//I GN + F F F G G G G G G G G G G G G G G G G G Gs4AA!$A!r3)rrr)rrr)rrrr)rrrnrn)rrrr)rrrr)rrrr)rrrr)r4r4rgrg)rnrnrr)r4r4rr)rrrr)rrrr)rrrrcd|D}|jj||}t|||dkrt jt |jt |ft}|ddd\}} d||| f<|jj|||}|d kr|}t||dSdS) zQTests the `query` method with valid inputs and valid predicates. c g|] }t| SrN)r )rWrs r(rZz8TestShapelyInterface.test_query_bulk..s777Tc4j777r*rrr,Nr4Trrr) r#rrrr test_geomsrrrothers r(test_query_bulkz$TestShapelyInterface.test_query_bulk6sT87Y777 gn"":"CC3))) I % %Hc$'llC OO.Es ) ) )1eAhh ) ) )r*r.) r0r1rr2r&r r r3rr r_points_from_xy) r#rrr+r4r`psrr'rXrYgas r(test_nearest_multiz'TestShapelyInterface.test_nearest_multi9s9ry}}bimm<<  #Z$7 8 8 ) ) ) ) )iz::3))) ^H % %iz::3)))   # #ij993)))H~1  %a + +iz::3)))))r*ctjtjdtjd}t jd|i}|j||}t||dSr-) r0r1rr2r&r r r3r)r#rrr+r4r`rs r(test_nearest_nonez&TestShapelyInterface.test_nearest_noneVsjry}}bimm<<  #Z$7 8 8iZ@@3)))))r*return_distancez return_all,max_distance,expected)rrrR2ctjtjdtjd}t jd|i}t ddt ddg}|j||||}|r:t|d|dt|d|ddSt||ddS)Nrnrrr)r+ max_distancer?rr/) r#rrBr+r?r4r`r:rs r(test_nearest_max_distancez.TestShapelyInterface.test_nearest_max_distanceesry}}bimm<<  #Z$7 8 8CoouQ|| ,i !%+      1 s1vx{ 3 3 3 s1vx{ 3 3 3 3 3 sHQK 0 0 0 0 0r*z*return_all,max_distance,exclusive,expected)rrrgr>rf)rrrrgr>)rrrrgrgr>r>rf)rrrgrr>rgrfr>g?)rrgrR)rRrRrr>)rrgrRrR)rRrRrrgrfctjtjdtjd}|r$tj|t ddg}t jd|i}|}|j |||||} |r:t| d|dt| d|ddSt| |ddS)NrRrrgr)r+rBr? exclusiver) r0r1rr2appendrr&r r r3r) r#rrBr+r?rFr4r`r:rs r(test_nearest_exclusivez+TestShapelyInterface.test_nearest_exclusives(ry||RYq\\::  4IeeAqkk]33E  #Z$7 8 8 i !%+      1 s1vx{ 3 3 3 s1vx{ 3 3 3 3 3 sHQK 0 0 0 0 0r*ctdddttddtg}tj|}|jtddddksJdS)z8Tests building sindex with interleaved empty geometries.rNrrr>)rr&r r r)r#r4r`s r(test_empty_tree_geometriesz/TestShapelyInterface.test_empty_tree_geometriessoq!dEGGU1a[[%''B  #U 3 3 3yuQ{{++A.!333333r*c>|jjj|jksJdS)zTests the `size` property.N)r`r r6rrks r( test_sizezTestShapelyInterface.test_sizes#w~"d&8888888r*cNt|jj|jksJdS)z.Tests the `__len__` method of spatial indexes.N)r/r`r rrks r(test_lenzTestShapelyInterface.test_lens(47>""d&8888888r*c^tjgt}|jjsJtjdg}|jjsJtjt g}|jjsJtjt ddg}|jjrJdS)zTests the `is_empty` property.r,Nr)r&r r.r is_emptyr)r#empty non_emptys r( test_is_emptyz"TestShapelyInterface.test_is_emptys#Bf555|$$$$#TF++|$$$$#UWWI..|$$$$'q! 66 #,,,,,,r*zpredicate, expected_shape))N)rgi)rrg)rrT)rrgr)rrV)rrV)rrVct|}t|}|j|j|}|j|ksJdS)z1Tests output sizes for the naturalearth datasets.N)r r rrshape)r#rexpected_shapenaturalearth_lowresnaturalearth_citiesworldcapitalsrs r(test_integration_natural_earthz3TestShapelyInterface.test_integration_natural_earthsS -..011l  !2I>>yN******r*)0rKrLrMrcrmark parametrizerrrparam SCIPY_MARKr rrrskipifrx GEOS_GE_310rrrrrrrrrrrrrrrrr#r(r*r6r<r>rCrHrJrLrNrSr^rNr*r(rrs/// [ !2 & #aS ) Aq6 " aSM  ***  [[>4uuQPQ{{*STT33UT3 [)\V\(*%M%M%MwW [( 33r2tT**B / 33tT3,,qc 2 33q!Q??QF + FF+,,A  33r2tT22B 7D$S))  Aq!QA   FF+,,  ss4tT22B 7 ss1aR((" - ss2r2r**QC 0 zz66"233R 8 Q1a" - Q5%001# 6 S#sC001# 6 RQ**QF 3 FF+,,  BAq))A3 /D$S))  BB##  D$1%%A  E5$--  Aq!QA  $Aq!Q  $Aq%''  $CeU++  $Cc3''  $BAq!!A  $BB## s^ aaD++EaaJ+ 222  [F..7KLL ['Cc3'' Cc3'' c3 Z  4'c3A Q$G44455G44455|||,  Q$G44455G44455'&&(=(=(=>  BsC%%1++.&&&(=(=(=> Y1 44j**k44MLl*  [F..7KLLCCMLC [ OOO [#UOOO [# 2J   ! !2 & UWWbM \^^R WYYO    00  0 >>>  [ IIIyyy) * YYY * + !!!FGGG  [)\V\(*%M%M%MwW [( ()B8 4 *+qcA3Z 8 L>QFQF#3 4 01B8 < 23qcA3Z @ L>QFQF+; < 02HIQCRSQT: V!78III&  01B8 < '"b 2 ()QC!: 6R  %&qc   )*qc   ./1#s < Q!Q   !"qc   (A3* 5'(R  #$Q!Q   +,R  Q!Q   $R  $%&qc   $)*qc   $%&qc   $ Q!Q   $!"R u_ bbF++GbbL+( [$! "" #b"X .  ""D )B8 4Vb"X CCdC--t 4sQCj A    00  0 000---UUU [( '"b 2 /"b : 1#s 4 ***6 [ IIIyyy) * YYY * +    F [\D%=99 [D\QC!: &D\QC!: &  * *:9 * [\D%=99 [f AA/ 0i Aq6Aq6"2 3 **:9** [\D%=99 [ B8 Vb"X  **:9* [.u >> [* 4999iii04499dd3iib2RS T DQFQF+dd3iib-BC D 1AA'$$s))TT#YY)?@ A A!qc TT#YYK0 1  11?>1& [.u >> [4 D%???OOO"Da1#g!N O D$///???!CQ$$q''] S T 4///???!CQ!W M N***,D,D,DEqDDQRGG9}U   CIII 6QC@ A 3 lll;Q!WE F  11!?>"14444 999999 - - - [#     ++  +++r*r)"mathrnumpyrr0shapely.geometryrrrrrr r&r r r r rxr numpy.testingr scipy.sparser HAS_SCIPY ImportErrorr_rcrbrrPskip usefixturesrrrNr*r(rns)8888888888'''''' ,,,,,,&&&&&&IIIII [  I 6K  L L J3J3J3J3J3J3J3J3Z[([([([([([([([(~566 * * * * * * *76 *Fl +l +l +l +l +l +l +l +l +l +sAAA