rbi3J dZddlmZddlmZddlmZmZddlZ ddl m Z m Z m Z ddlmZmZmZmZmZmZddlmZdd lmZmZddlZeejd z Zd d d d dddddd Zde DZ!gdZ"dddddddZ#de vZ$dZ%ej&'e dZ(ej&'e dZ)ej&'e pe dZ*ej&'e dZ+ej&'e dZ,dZ-ej.dd Z/ej.d!d"Z0ej.d!e"#d$Z1ej.d!d%g#d&Z2ej.d!d'Z3ej.d!d(Z4ej.dd)Z5ej.dd*Z6ej.dd+Z7ej.dd,Z8ej.dd-Z9ej.d!d.Z:d/Z;d0Z<ej.d!d%d1g#d2Z=ej.d!d3Z>ej.d!d4Z?ej.d!d5Z@ej.d!d6ZAej.d!d7ZBej.d!d8ZCej.d!d9ZDej.d!d:ZEej.d!d;ZFej.d!d<ZGej.dgd=#d>ZHdS)?zOModule with helper functions, fixtures, and common test data for pyogrio tests.)BytesIO)Path) ZIP_DEFLATEDZipFileN)__gdal_version_string__ __version__ list_drivers) GDAL_GE_37HAS_ARROW_WRITE_API HAS_GDAL_GEOS HAS_PYARROW HAS_PYPROJ HAS_SHAPELY) vsi_rmtree)readwritefixtures FlatGeobufGeoJSON GeoJSONSeqGPKGzESRI Shapefile) .fgb.geojson .geojsonl .geojsons.gpkg .gpkg.zip.shpz.shp.zipz.shzci|]\}}|| Sr ).0extdrivers f/var/www/html/mdtn/previsions/meteo_cartes/venv/lib/python3.11/site-packages/pyogrio/tests/conftest.py r%'s = = =kc6fc = = =)rrrrr)rrrrrrParquetcddttD}dtdt d|S)Nz, c3,K|]\}}|d|dVdS)()Nr )r!r# capabilitys r$ z'pytest_report_header..8sK FJ !!J!!!r&zpyogrio z GDAL z Supported drivers: )joinsortedr itemsrr)configdriverss r$pytest_report_headerr47syii"()=)=)?)?"@"@G  (; ( (' ( (% ( (r&zpyarrow required)reasonzpyproj requiredz&GDAL>=3.8 required for Arrow write APIz GDAL compiled with GEOS requiredzShapely >= 2.0 requiredc.|dkrtstjd||jkr|S||j|z }|r|St |\}}}}|dkr d|d<d|d<n|dvr d|d<d |d<t|||fi||S) Nrz&.gpkg.zip support requires GDAL >= 3.7rF spatial_indexUnknown geometry_type)rr MultiPolygon)r pytestskipsuffixstemexistsrr) testfile_pathdst_dirr"dst_pathmeta_geometry field_datas r$prepare_testfilerGTs k* <=== m"""M.5555H$($7$7!D!Xz f}} %_ )_ & & & %_ ._ (Hj11D111 Or&session)scopectSN _data_dirr r&r$data_dirrNps r&functionctt|dd}ttdz }t|||S)Nparamr+naturalearth_lowres/naturalearth_lowres.shp)getattrrMrrG)tmp_pathrequestr"r@s r$naturalearth_lowresrVus8 '7F + +C%R S SSM M8S 9 99r&)rIparamsc.t|||jSrKrGrQrTrVrUs r$naturalearth_lowres_all_extr[} /7= I IIr&rc.t|||jSrKrYrZs r$naturalearth_lowres_geojsonr^r\r&c||jdz }t|dtd5}dD],}|jd|}||j|z |- dddn #1swxYwY|d|d |jfS) z4Wrap naturalearth_lowres as a zip file for VSI testsz.zipw)mode compression compresslevel)dbfprjshpshxcpg.Nz/vsizip//)namerrr>rparent)rTrVpathoutr"filenames r$naturalearth_lowres_vsirqs ,1777 7D C\ K K KGs6 G GC-2::S::H II)08;X F F F F GGGGGGGGGGGGGGGG =D==#6#;== ==s0A##A'*A'c#Kt|\}}}}d|j}td|d|d}d|d<d|d<t|||fd d i||Vt |jd S) z8Write naturalearth_lowres to a vsimem file for VSI testspyogrio_fixture_z/vsimem/rkrFr7r:r9layerrVN)rr>rrrrm)rVrCrDrErFrlrBs r$naturalearth_lowres_vsimemrus%))<$=$=!D!Xz 816 8 8D1t11d11122H!D*D (HjNN0ENNNN NNNxr&ctdz S)Nz line_zm.gpkgrLr r&r$ line_zm_filerws ~ %%r&ctdz S)Nz curve.gpkgrLr r&r$ curve_filerys | ##r&ctdz S)Nzcurvepolygon.gpkgrLr r&r$curve_polygon_filer{ * **r&ctdz S)Nzmultisurface.gpkgrLr r&r$multisurface_filer~r|r&ctdz S)Nztest_gpkg_nulls.gpkgrLr r&r$test_gpkg_nullsrs - --r&c d|dz }t|ddtjgdgdg|S)Nztest_no_geometry.gpkg no_geometry)abccol)rtrErFfields)rnparray)rTrps r$no_geometry_filersO11H H___--.w  Or&cd}|dz }t|d5}||}dddn #1swxYwY|S)Na { "type": "FeatureCollection", "features": [ { "type": "Feature", "properties": { "int": 1, "list_int": [0, 1], "list_double": [0.0, 1.0], "list_string": ["string1", "string2"], "list_int_with_null": [0, null], "list_string_with_null": ["string1", null] }, "geometry": { "type": "Point", "coordinates": [0, 2] } }, { "type": "Feature", "properties": { "int": 2, "list_int": [2, 3], "list_double": [2.0, 3.0], "list_string": ["string3", "string4", ""], "list_int_with_null": [2, 3], "list_string_with_null": ["string3", "string4", ""] }, "geometry": { "type": "Point", "coordinates": [1, 2] } }, { "type": "Feature", "properties": { "int": 3, "list_int": [], "list_double": [], "list_string": [], "list_int_with_null": [], "list_string_with_null": [] }, "geometry": { "type": "Point", "coordinates": [2, 2] } }, { "type": "Feature", "properties": { "int": 4, "list_int": null, "list_double": null, "list_string": null, "list_int_with_null": null, "list_string_with_null": null }, "geometry": { "type": "Point", "coordinates": [2, 2] } }, { "type": "Feature", "properties": { "int": 5, "list_int": null, "list_double": null, "list_string": [""], "list_int_with_null": null, "list_string_with_null": [""] }, "geometry": { "type": "Point", "coordinates": [2, 2] } } ] }ztest_ogr_types_list.geojsonr`openr)rT list_geojsonrpfrDs r$list_field_values_geojson_filers@ LD77H h  " GGL ! !""""""""""""""" O :>>c tdz }|r|S ddl}ddlm}ddl}n(#t $r}t d|d|dd}~wwxYw||| tj dgd dd gd d ggddgd dgddggddgddggdgddggddgd d ggddgddggdgddggd}| |||S)aEReturn the path to a Parquet file with list values in a property. Because in the CI environments pyarrow.parquet is typically not available, we save the file in the test data directory instead of always creating it from scratch. The code to create it is here though, in case it needs to be recreated later. zlist_field_values_file.parquetrNparquet test file & does not exist, but error importing: rj)ra)r'rrar'rrgg?g@g@string1string2)string3string4r)rEintlist_int list_double list_stringlist_int_with_nulllist_string_with_null rMr?pyarrowrshapely ImportError RuntimeErrortableto_wkbpointsrones write_table) fixture_pathpapqrexrs r$list_field_values_parquet_filers??L ))))))     R R RR R R R    HHw~~bgfoo'F'FGG"??Q!QT48 #Jc BdCI&*** %&t9q!fb$"ED!*** &    E.NN5,''' 1 AAA.parquetcl|jdkrt|S|jdkrtSdS)Nrr)rQrr)rTrUs r$list_field_values_filesrKs>} ""-h777 * $ $-/// % $r&cd}|dz }t|d5}||}dddn #1swxYwY|S)Na.{ "type": "FeatureCollection", "features": [ { "type": "Feature", "geometry": { "type": "Point", "coordinates": [0, 0] }, "properties": { "int_col": 1, "float_col": 1.5, "str_col": "string", "bool_col": true, "null_col": null, "date_col": "2020-01-01", "time_col": "12:00:00", "datetime_col": "2020-01-01T12:00:00", "list_int_col": [1, 2, 3], "list_str_col": ["a", "b", "c"], "list_mixed_col": [1, "a", null, true] } } ] }ztest_many_data_types.geojsonr`r)rTmany_types_geojsonrprrDs r$many_data_types_geojson_filerSs 488H h  ( GG& ' '((((((((((((((( Orcd}|dz }t|d5}||}dddn #1swxYwY|S)Na{ "type": "FeatureCollection", "features": [ { "type": "Feature", "geometry": { "type": "Point", "coordinates": [0, 0] }, "properties": { "top_level": "A", "intermediate_level": { "bottom_level": "B" } } } ] }ztest_nested.geojsonr`r)rTnested_geojsonrprrDs r$nested_geojson_filerws N&//H h  $ GGN # #$$$$$$$$$$$$$$$ Orc tdz }|r|S ddl}ddlm}ddl}n(#t $r}t d|d|dd}~wwxYw||| tj dgd d d d gd zd d d gd zgd zgdgd zd}| |||S)aGCreate a Parquet file in tmp_path with nested values in a property. Because in the CI environments pyarrow.parquet is typically not available, we save the file in the test data directory instead of always creating it from scratch. The code to create it is here though, in case it needs to be recreated later. zlist_nested_struct_file.parquetrNrrrrj)rr)rr'rr'r)rrr)r'rr)rEcol_flat col_struct col_nestedcol_listr)rTrrrrrrs r$list_nested_struct_parquet_filers>@@L ))))))     R R RR R R R    HHw~~bgfoo'F'FGG! !"++,q0"#!,,-12Q6" a     ENN5,''' rcd}|dz }t|d5}||}dddn #1swxYwY|S)Na{ "type": "FeatureCollection", "features": [ { "type": "Feature", "properties": { "col": "2020-01-01T09:00:00.123" }, "geometry": { "type": "Point", "coordinates": [1, 1] } }, { "type": "Feature", "properties": { "col": "2020-01-01T10:00:00" }, "geometry": { "type": "Point", "coordinates": [2, 2] } } ] }ztest_datetime.geojsonr`r)rTdatetime_geojsonrprrDs r$ datetime_filers  11H h  & GG$ % %&&&&&&&&&&&&&&& Orcd}|dz }t|d5}||dddn #1swxYwY|S)Na{ "type": "FeatureCollection", "features": [ { "type": "Feature", "properties": { "datetime_col": "2020-01-01T09:00:00.123-05:00" }, "geometry": { "type": "Point", "coordinates": [1, 1] } }, { "type": "Feature", "properties": { "datetime_col": "2020-01-01T10:00:00-05:00" }, "geometry": { "type": "Point", "coordinates": [2, 2] } } ] }ztest_datetime_tz.geojsonr`rrTdatetime_tz_geojsonrprs r$datetime_tz_filers  44H h  % #$$$%%%%%%%%%%%%%%% Orctttdz d\}}}}|dz }t|||fi|t |d5}|}dddn #1swxYwY|S)z\Extracts first 3 records from naturalearth_lowres and writes to GeoJSON, returning bytesrRr max_features test.geojsonrbNrrMrrr)rTrCrDrErFrpr bytes_buffers r$ geojson_bytesrs%)DFGGGVW%%%!D!Xz.(H (Hj11D111 h   vvxx                 sA11A58A5cd}|dz }t|d5}||dddn #1swxYwY|S)Na{ "type": "FeatureCollection", "features": [ { "type": "Feature", "properties": { "datetime_col": "1670-01-01T09:00:00" }, "geometry": { "type": "Point", "coordinates": [1, 1] } } ] }ztest_datetime_long_ago.geojsonr`rrs r$geojson_datetime_long_agors  ::H h  % #$$$%%%%%%%%%%%%%%% Orc#Ktttdz d\}}}}|dz }t|||fddi|t |d5}|VddddS#1swxYwYdS) zgExtracts first 3 records from naturalearth_lowres and writes to GeoJSON, returning open file handlerRrrrrttestrNr)rTrCrDrErFrprs r$geojson_filelikers%)DFGGGVW%%%!D!Xz.(H (Hj???$??? h  sA&&A*-A*cd}|dz }t|d5}||}dddn #1swxYwY|S)Na interfaces1 Ton 19.1501280458077,293.313485355882 ztest.kmlr`r)rTkml_datarprrDs r$kml_filer$sH$*$H h   GGH   OrchGddt}d}||dS)NceZdZdZdZdS)-nonseekable_bytes..NonSeekableBytesIOcdS)NFr )selfs r$seekablez6nonseekable_bytes..NonSeekableBytesIO.seekableDs5r&c td)Nz cannot seek)OSError)rargskwargss r$seekz2nonseekable_bytes..NonSeekableBytesIO.seekGs-(( (r&N)__name__ __module__ __qualname__rrr r&r$NonSeekableBytesIOrCs2    ) ) ) ) )r&rz{ "type": "FeatureCollection", "features": [ { "type": "Feature", "properties": { }, "geometry": { "type": "Point", "coordinates": [1, 1] } } ] }zUTF-8)rencode)rTrgeojsons r$nonseekable_bytesr@sR)))))W)))  G  gnnW55 6 66r&))CP932uﾎ)CP936u中文)CP1250uĐ)CP1252ÿ)CP1253uΦ)CP1256uشc|jS)a'Return tuple with encoding name and very short sample text in that encoding NOTE: it was determined through testing that code pages for MS-DOS do not consistently work across all Python installations (in particular, fail with conda), but ANSI code pages appear to work properly. )rQ)rUs r$ encoded_textrYs . =r&)I__doc__iorpathlibrzipfilerrnumpyrpyogriorrr pyogrio._compatr r r r rr pyogrio.corer pyogrio.rawrrr;__file__rmresolverMDRIVERSr1 DRIVER_EXTALL_EXTS START_FIDGDAL_HAS_PARQUET_DRIVERr4markskipifrequires_pyarrow_apirequires_pyprojrequires_arrow_write_apirequires_gdal_geosrequires_shapelyrGfixturerNrVr[r^rqrurwryr{r~rrrrrrrrrrrrrrrrr r&r$rsUU))))))))FFFFFFFFFF$############# DNN ! ) ) + +j 8       > =W]]__ = = = = = =      $||~~5   {))k/BT)UU+$$^W%XX8i   ! j!!!::"!:j222JJ32Jj*666JJ76Jj!!! > >"! >j!!!   "!  i   &&! &i   $$! $i   ++! +i   ++! +i   ..! .j!!!  "! HHHV111hj*j)ABBB00CB0j!!!  "! Fj!!!"!8j!!!##"!#Lj!!!"!2j!!!"!2j!!!  "!  j!!!"!(j!!!  "! j!!!"!6j!!!77"!70    "#"r&