hddlmZddlZddlZddlmZmZmZm Z m Z m Z ddl Z ddl m Z mZmZddl mZddlZddlmZ ddlmZdZej2j5e d ZGd d ZGddZej2j<ej2j?dGddZ GddZ!y#e$rd ZYwxYw))sqrtN)GeometryCollection LineString MultiPolygonPointPolygonbox) GeoDataFrame GeoSeries read_file)_compat)assert_array_equal) coo_arrayTFzscipy not installedreasonc<eZdZdZdZdZdZdZdZdZ dZ y ) TestSeriesSindexc tgd}tgd}td||gid}|jrJ|j|jsJd|jj _|jrJ|j|jsJt||g}|jrJ|j|jsJd|j _|jrJ|j|jsJy)zTest the has_sindex method.rrrrrrrrrgeomgeometryN)rr has_sindexsindexrvalues_sindexr )selft1t2dss N/var/www/html/immo/lib/python3.12/site-packages/geopandas/tests/test_sindex.pytest_has_sindexz TestSeriesSindex.test_has_sindex s - . - . &2r(+f =<< |||$( !<< ||| r2h << |||<< |||cttt}|jrJt|jdk(sJy)z7Tests creating a spatial index from an empty GeoSeries.dtyperN)r objectr lenr#r's r(test_empty_geoseriesz%TestSeriesSindex.test_empty_geoseries7s. F #88|188}!!!r*c6ttddg}|jjdk(sJ|jj d}t t |dk(sJ|jj d}t t |dk(sJy)Nrrr4rr)r5r4r4)r rr size intersectionr/list)r#r'hitss r( test_pointzTestSeriesSindex.test_point=s uQ{m $xx}}!!!xx$$^44:!###xx$$%564:!###r*c|ttg}|jrJt|jdk(sJy)z9Tests that a single empty Point results in an empty tree.rN)r rr r/r0s r(test_empty_pointz!TestSeriesSindex.test_empty_pointEs2 uwi 88|188}!!!r*ctgd}tgd}tgd}t|||g}|jjdk(sJy)Nrr)rrrr)rr r r6)r#r$r%sqr's r( test_polygonszTestSeriesSindex.test_polygonsKsI - . - . 5 6 r2rl #xx}}!!!r*cttddg}|jjJ|jj dk(sJ|jjJy)Nrr)r rr!r"r r6r0s r(test_lazy_buildz TestSeriesSindex.test_lazy_buildRsV uQ{m $xx'''xx}}!!!xx+++r*cttddg}|j}tdd|jd<|j|usJy)Nrr rr iloc)r#r'original_indexs r(test_rebuild_on_item_changez,TestSeriesSindex.test_rebuild_on_item_changeXsB uQ{m $!QKq xx~---r*cttddtddg}|j}|jdd}|j|usJ|jdd}|j|usJ|jddd}|j|usJy)Nrrr4rD)r#r'rFsliceds r(test_rebuild_on_slicez&TestSeriesSindex.test_rebuild_on_slice^s uQ{E!QK0 1}}N222}}..."}}N222r*N) __name__ __module__ __qualname__r)r1r:r<r@rBrGrJr*r(rrs*." $" ", . 3r*rcBeZdZdZdZdZdZdZdZdZ dZ d Z y ) TestFrameSindexc tdtddttdtdDcgc]\}}t||c}}d}t|d|_ycc}}w)Nr)ABrrr)rangeziprr dfr#xydatas r( setup_methodzTestFrameSindex.setup_methodmsVqr1-0q58-DETQU1a[E  tf5FsA& cd|j_|jjjdk(sJt |jjj d}t |dk(sJ|ddk(sJy)Nz epsg:4326rR)@r_r`rr>)rXcrsr r6r8r7r/)r#r9s r( test_sindexzTestFrameSindex.test_sindexusg! ww~~""a'''DGGNN//0@AB4yA~~Aw!||r*c|jjjjJ|jjj dk(sJ|jjjjJy)NrR)rXrr!r"r r6r#s r(rBzTestFrameSindex.test_lazy_build|sbww&&..666ww~~""a'''ww&&..:::r*c P|jjJ|jj}|jjtt ddt ddDcgc]\}}t ||c}}d|jj|usJycc}}w)NrR T)inplace)rXr set_geometryrWrVr)r#rFrZr[s r(#test_sindex_rebuild_on_set_geometryz3TestFrameSindex.test_sindex_rebuild_on_set_geometrysww~~))) %(q"uQ|%D ETQU1a[ Et  ww~~^333 Fs'B" cX|jj}|jjdd}|j|usJ|jj}|jjdd}|j|usJ|jjddd}|j|usJy)Nrr4)rXr rE)r#rFrIs r(test_rebuild_on_row_slicez)TestFrameSindex.test_rebuild_on_row_slicesbq!}}N222a}}...dd#}}N222r*c|jj}|jd}|j|usJ|jj}|j|usJy)z?Selecting a single column should not rebuild the spatial index.rN)rXr r)r#rF geometry_cols r($test_rebuild_on_single_col_selectionz4TestFrameSindex.test_rebuild_on_single_col_selectionsUwwv ""n444ww'' ""n444r*c6|jj}|jddg}tjs|j|usJ|j|usJ|jddg}tjs|j|usJy|j|usJy)z2Selecting a subset of columns preserves the index.rrTN)rXr compat PANDAS_GE_30)r#rFsubset1subset2s r(&test_rebuild_on_multiple_col_selectionz6TestFrameSindex.test_rebuild_on_multiple_col_selections''63-("">>7 77>>^3 33''3-("">>7 77>>^3 33r*c|jj}|j}|jddd|jrJ|j}||usJ|j j gdk(sJy)NrTFT) ascendingrh)r`r>rarr)rXcopyr sort_valuesrindextolistr#gdf old_sindex new_sindexs r(test_rebuild_on_update_inplacez.TestFrameSindex.test_rebuild_on_update_inplacesmggllnZZ  ud;>>!!ZZ +++yy!_444r*c|jj}|j}|jddid|jsJ|j}||usJy)NrTAAT)columnsrh)rXrxr renamerr|s r(test_update_inplace_no_rebuildz.TestFrameSindex.test_update_inplace_no_rebuildsQggllnZZ  C; 5~~~ZZ Z'''r*N) rKrLrMr]rcrBrjrlrorurrrNr*r(rPrPls/6; 4 354" 5(r*rP_setup_class_nybb_filenameceZdZdZdZy)TestJoinSindexc8t|j|_y)N)r nybb_filenameborosres r(r]zTestJoinSindex.setup_methodst112 r*ct|jj}|jd}|Dcgc]}|jj|d }}|ddgk(sJ|j|jddk}|j}|jd}|Dcgc]}|j|d}}|dgk(sJ|j|jddk\}|j}|jd}|Dcgc]}|j|dc}f}|dgk(sJ|j |d}t |d k(sJ|jj d k(sJ|j}|jd}|Dcgc]}|j|d}}|ddgk(sJycc}wcc}wcc}wcc}w) N)g.AgHzb ABoroNameBronxQueensBoroCoder>outer)howrR)rr r7rEmerger/r6)r#treer9hitresfirstsecondmergeds r(test_merge_geozTestJoinSindex.test_merge_geoszz    !89;?@Ctzzs#J/@@w)))) 4::j1A56||  !896:;suzz#z*;;wiDJJz2a78}}  !898<= C ,=?xj   V16{a}}!!Q&&&}}  !897;<v{{3 +<<w))))1A<>=s#F&F+=F0F5N)rKrLrMr]rrNr*r(rrs 3*r*rc'eZdZdZej j ddgfddgfdddgfddgffd Zej j d d d d eddfdZ ej j ddejde dfej j dd e ddd d gfd e d d dddgfd e ddddddgfd e ddgddgfde ddd d gfde d d dddgfde ddddddgfde ddggfde ddddgfde ddddgfde ddd d d!gfde ddggfd"e ddddgfd"e ddd#d#dgfd"e ddd$d$dgfd"e ddd%d%ddgfd"e ddggfd&e dddddgfd&e d d d$d$gfd"e ddd'd'd!gfd(e d d ddddgfd(e d)d)d*d*gfd(e ddddddgfd+e ddddgfd+e ddd#d#dgfd+e ddd#d#dgfd+e ddd$d$dgfd+e ddd%d%ddgfd+e ddd'd'gffd,Zd-Zej j!ej$ d./ej j d0de d1d1d2d2gfde d1d1d2d2d!gfdeddgfedd3zeddddgfed%d3z egd4egd5ggdgd6gfed%d3zegd4egd5ggd7gd8gfd%dgeddeddggd9gd:gffd;Zej j!ej$ d./d<Zej j d=gd>d?Zej j!ej$d@/dAZej j dBd gfegfegfegfegfgdCZdDZej j dEdFgdGgdHgfdIgdGgdHgffdJZdKZej j ddejde dfej j dd dggggfd dgdgdggfd dgddgddggfddggggfddgdgdggfddgddgddggfdddgdgdggfddLdggdMgdNgfddOggggfddPggggfddQgdgd!ggfd"dggggfd"dRgdgdggfd"dSgdgdggfd"dTgdgdggfd"dUgddgddggfd"dVgdgd!ggfd&dWgdgdggfd&dXggggfd(dYgddgddggfd(dZggggfd(dgddgddggfd+dggggfd+dRgdgdggfd+dSgdgdggfd+dTgdgdggfd+dUgddgddggfd+dVggggffd[Zej j d\eggggfed ggggfd ggggfd e d d ddd gdgdggfgd]Z d^Z!d_Z"d`Z#ej j dd dgggfddgggfd"dLdgdggffdaZ$ej j dEdFgdGgdHgfdIgdGgdHgffdbZ%ej j dcdFdIgej j ddddgdgdggfddgdgdggfgdeZ&ej j dcdFdIgej j dddfdgddgddggfdfdggddgddggfgdhZ'ej j dcdFdIgej j ddd gggfd ggggfgdiZ(ej j djdFdIgej j dkdFd gdMgdlgedededmgffdId ddgdd!ggededmgffdFdddgddggededgffdIddgdggedgffgdnZ)ej j djdFdIgej j dodId dIgdpgdpgd!dgzffdId dFgdpgdqgd!ed%gzffdFd dIgdpgdpgd!dgzffdFd dFgdrgdsgdted%gzffdIdudFgdvgdwgdxdgzffdFdudFgdygdzgd{dgzffgd|Z*d}Z+d~Z,dZ-dZ.ej j dgddZ/y )TestShapelyInterfacec dttdtdDcgc]\}}t||c}}tddddgzi}t |d|_t |d|_ycc}}w)NrrRrgr)rWrVrr r rXr/ expected_sizerYs r(r]z!TestShapelyInterface.setup_methodso Sq58-DETQU1a[E2r2r"#$ tf5 f. FsA4ztest_geom, expectedr4r4r)rr?rr)rrrrrrcxt|jjj|}t ||y)z2Tests the `intersection` method with valid inputs.N)r8rXr r7r)r# test_geomexpectedrs r(test_intersection_bounds_tuplez3TestShapelyInterface.test_intersection_bounds_tuples+477>>..y9:3)r*r)r4r4rrNctjt5|jjj |dddy#1swYyxYw)z4Tests the `intersection` method with invalid inputs.N)pytestraises TypeErrorrXr r7r#rs r(&test_intersection_invalid_bounds_tuplez;TestShapelyInterface.test_intersection_invalid_bounds_tuples;]]9 % 3 GGNN ' ' 2 3 3 3 &A  A output_formatindicessparse)marksdensezpredicate, test_geom, expectedr4rrr intersectswithin?Q??rg rRcontainsjt??ratouchesrcoversMbP?Gz?contains_properlycn|jjj||}t|||dk7r}t j t |jt}d||<|jjj|||}|dk(r|j}t||yy)z@Tests the `query` method with valid inputs and valid predicates. predicaterr,TrrrN) rXr queryrnpzerosr/booltodense)r#rrrrrrs r( test_queryzTestShapelyInterface.test_querysNggnn""9 "B3) I %HHS\6E"E(O''..&&Ym'C(kkm sE * &r*ctjt5|jjj ddddy#1swYyxYw)z/Tests the `query` method with invalid geometry. notavalidgeomNrrrrXr rres r(test_query_invalid_geometryz0TestShapelyInterface.test_query_invalid_geometrys9 ]]9 % 2 GGNN  1 2 2 2rzRequires GEOS 3.10rzdistance, test_geom, expectedg"@g#@g& .>r)r)rar)rara)rrrra)rrrrrrr)rrrarrrar>)rrrrrrr)rrrrrar>r`cl|jjj|d|}t||y)zHTests the `query` method with predicates that require keyword arguments.dwithinrdistanceNrXr rr)r#rrrrs r(test_query_dwithinz'TestShapelyInterface.test_query_dwithins.pggnn""9 H"U3)r*ctjtd5|jjj t ddddddy#1swYyxYw)`Tests the `query` method with keyword arguments that are invalid for certain predicates.z8'distance' parameter is required for 'dwithin' predicatematchrrrNrr ValueErrorrXr rrres r(test_dwithin_no_distancez-TestShapelyInterface.test_dwithin_no_distancesR]] X  C GGNN q! B C C Cs 2AA r) Nrr covered_byrcrossesroverlapsrrcd}tjt|5|jjj t dd|ddddy#1swYyxYw)rzD'distance' parameter is only supported in combination with 'dwithin'rrrNr)r#rmsgs r(test_query_distance_invalidz0TestShapelyInterface.test_query_distance_invalidsT$U ]]:S 1 O GGNN q! A N O O Os 3AA#z0Test for 'dwithin'-incompatible versions of GEOSctjtd5|jjj t dddddddy#1swYyxYw)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.0rrrrNrres r(test_dwithin_requirementsz.TestShapelyInterface.test_dwithin_requirementssT]] M  O GGNN q! A N O O Os 3AA!ztest_geom, expected_valuecf|jjj|}t||y)z-Tests the `query` method with empty geometry.Nr)r#rexpected_valuers r(test_query_empty_geometryz.TestShapelyInterface.test_query_empty_geometrys&ggnn""9-3/r*ctdddd}tjt5|jj j |ddddy#1swYyxYw)1Tests the `query` method with invalid predicates.r4rtestrN)r rrrrXr rrs r(test_query_invalid_predicatez1TestShapelyInterface.test_query_invalid_predicatesNBd+ ]]: & > GGNN f = > > >s (AA"zsort, expectedT)rrrrrraFcttgdg}ttgdtgdtgdg}gd}|jd}|jj ||}t |t |k(sJ t ||y #t$r5}|dur*tjd|d |jd |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_sortings(H IJK 89:;89  $$Q'%%hT%:c{fX....  sH - u} ##+*-!jjl^2/ G s B C'0CCctjtd5tdddd}|jj j |ddddy#1swYyxYw)Nz#Invalid output_format: 'dataarray'.rr4r dataarray)r)rrrr rXr rrs r(test_unsupported_outputz,TestShapelyInterface.test_unsupported_output0sU ]]:-R S GBD$/I GGNN + F G G Gs 6AA$r3)rrr)rrr)rrrr)rrrgrg)rrrr)rrrr)rrrr)rrrr)r4r4rara)rgrgrr)r4r4rr)rrrr)rrrr)rrrrc|Dcgc] }t| }}|jjj||}t |||dk7rt j t|jt|ft}|ddd\} } d|| | f<|jjj|||}|dk(r|j}t ||yycc}w) zQTests the `query` method with valid inputs and valid predicates. rrr,Nr4Trr) r rXr rrrrr/rr) r#rrrrr test_geomsrrrothers r(test_query_bulkz$TestShapelyInterface.test_query_bulk6sT.77Tc4j7 7ggnn"":"C3) I %HHc$''lC Or`)rrrrar>)rrrrarar>r>r`)rrrarr>rar`r>g?)rrarR)rRrRrr>)rrarRrR)rRrRrrar`ctjtjdtjd}|r!tj|t ddg}t jd|i}|}|jj|||||} |r%t| d|dt| d|dyt| |dy)NrRrrar)rr0r- exclusiver) rr rr!appendrrr r r"r) r#rr0rr-r4r#rXr(rs r(test_nearest_exclusivez+TestShapelyInterface.test_nearest_exclusives(ryy|RYYq\: IIeeAqk]3E  # #Z$7 8 ii !%+     s1vx{ 3 s1vx{ 3 sHQK 0r*ctdddttddtg}tj|}|jj tddddk(sJy)z8Tests building sindex with interleaved empty geometries.rNrrr>)rrr r r)r#r#rXs r(test_empty_tree_geometriesz/TestShapelyInterface.test_empty_tree_geometriessXq!dEGU1a[%'B  # #U 3yyuQ{+A.!333r*cb|jjj|jk(sJy)zTests the `size` property.N)rXr r6rres r( test_sizezTestShapelyInterface.test_sizes$ww~~""d&8&8888r*c`t|jj|jk(sJy)z.Tests the `__len__` method of spatial indexes.N)r/rXr rres r(test_lenzTestShapelyInterface.test_lens#477>>"d&8&8888r*ctjgt}|jjsJtjdg}|jjsJtjt g}|jjsJtjt ddg}|jjrJy)zTests the `is_empty` property.r,Nr)rr r.r is_emptyr)r#empty non_emptys r( test_is_emptyz"TestShapelyInterface.test_is_emptys##Bf5||$$$$##TF+||$$$$##UWI.||$$$$''q! 6 ##,,,,,r*zpredicate, expected_shape))N)rai)rra)rrB)rrar)rrD)rrD)rrDct|}t|}|jj|j|}|j|k(sJy)z1Tests output sizes for the naturalearth datasets.N)r r rrshape)r#rexpected_shapenaturalearth_lowresnaturalearth_citiesworldcapitalsrs r(test_integration_natural_earthz3TestShapelyInterface.test_integration_natural_earthsH -.01ll  !2!2I>yyN***r*)0rKrLrMr]rmark parametrizerrrparam SCIPY_MARKr rrrskipifrq GEOS_GE_310rrrrrrrrrrrrrrr rrrrr%r*r,r1r6r8r:r<rArLrNr*r(rrs./ [[ !2 & #aS ) Aq6 " aSM  **  [[[>4uQPQ{*ST3U3 [[)\V\\(*%MwW [[( 3r2tT*B / 3tT3,qc 2 3q!Q?QF +FF+,A  3r2tT2B 7D$S)  Aq!QA  FF+,  s4tT2B 7 s1aR(" - s2r2r*QC 0 z66"23R 8 Q1a" - Q5%01# 6 S#sC01# 6 RQ*QF 3FF+,  BAq)A3 /D$S)  BB#  D$1%A  E5$-  Aq!QA  $Aq!Q  $Aq%'  $CeU+  $Cc3'  $BAq!A  $BB# s^ aD+EaJ+ 2  [[F...7KL [['Cc3' Cc3' c3 Z 4'c3A Q$4545|,  Q$4545'(=>  BsC%1+.&(=> Y1 4j*k4Ml*  [[F...7KLCMC [[ OO [[#UOO [[# 2J  !2 & WbM ^R YO   0 0 >  [[ Iy) * Y * + !!FG  [[)\V\\(*%MwW [[( ()B8 4 *+qcA3Z 8 L>QFQF#3 4 01B8 < 23qcA3Z @ L>QFQF+; < 02HIQCRSQT: V!78I&  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_ bF+GbL+( [[$!" #b"X . "D )B8 4Vb"X CdC-t 4sQCj A   0 0 0-U [[( '"b 2 /"b : 1#s 4 **6 [[ Iy) * Y * +   F [[\D%=9 [[D\QC!: &D\QC!: &  *: * [[\D%=9 [[f AA/ 0i Aq6Aq6"2 3 *:** [[\D%=9 [[ B8 Vb"X  *:* [[.u > [[* 49i049d3ib2RS T DQFQF+d3ib-BC D 1AA'$s)T#Y)?@ A A!qc T#YK0 1  1?1& [[.u > [[4 D%?O"Da1#g!N O D$/?!CQ$q'] S T 4/?!CQ!W M N*,DEqDQRG9}U   CI 6QC@ A 3 l;Q!WE F  1!?"144 99 - [[#  + +r*r)"mathrnumpyrrshapely.geometryrrrrrr rr r r r rqr numpy.testingr scipy.sparser HAS_SCIPY ImportErrorrMrQrPrrPskip usefixturesrrrNr*r(r\s88' ,&I[[  I 6K  L J3J3Z[([(~56 * *7 *Fl +l +mIsCC C