a 1$e@sddlZddlmZddlmZddlmZmZm Z ej ej e Zej ej eZzddlmZWneyddlZYn0dddZd d Zd d ZdddZdS)N)Series) GeoDataFrame)assert_geoseries_equalgeom_almost_equals geom_equals)mockFcCst|tsJt|dksJd}|r@|D]}||jvs*Jq*n&|D] }|dd|jDvsDJqDt|jj d sJdS)zATests a GeoDataFrame that has been read in from the nybb dataset.)BoroCodeBoroName Shape_Leng Shape_Areacss|]}|VqdSN)lower).0ZdfcolrP/var/www/html/django/DPS/env/lib/python3.9/site-packages/geopandas/tests/util.py $z#validate_boro_df..Z MultiPolygonN) isinstancerlencolumnsrrgeometry geom_typedropnaeqall)dfcase_sensitivercolrrrvalidate_boro_dfsrcCs|jdur|jpdSdS)zReturn srid from `df.crs`.Nr)ZcrsZto_epsg)rrrrget_srid(s r c s||jj}t||d|dd||jjdf|dd|fd}||fdd|j d d DWd n1s0Yd S) z Return a SpatiaLite connection containing the nybb table. Parameters ---------- `con`: ``sqlite3.Connection`` `df`: ``GeoDataFrame`` zCREATE TABLE IF NOT EXISTS nybb ( ogc_fid INTEGER PRIMARY KEY, borocode INTEGER, boroname TEXT, shape_leng REAL, shape_area REAL)z$SELECT AddGeometryColumn(?, ?, ?, ?)ZnybbrzSELECT CreateSpatialIndex(?, ?)z:INSERT INTO nybb VALUES(?, ?, ?, ?, ?, GeomFromText(?, ?))c3s6|].}d|j|j|j|j|jr&|jjndfVqdSr )r r r r rwkt)rrowsridrrrMs z$create_spatialite..F)indexN) rnamer executerrZiatupperZ executemanyZ itertuples)conrgeom_colZsql_rowrr#rcreate_spatialite/s$   r+geomc Cs|durd|}d|}nd}d}z|}|ddj||d}|||D]>\}} d |}||| dj| d | d | d | d fqZW||n||0dS)z Create a nybb table in the test_geopandas PostGIS database. Returns a boolean indicating whether the database table was successfully created Nzgeometry(MULTIPOLYGON, {})z'ST_SetSRID(ST_GeometryFromText(%s), {})rzST_GeometryFromText(%s)zDROP TABLE IF EXISTS nybb;zCREATE TABLE nybb ( {geom_col} {geom_schema}, borocode integer, boroname varchar(40), shape_leng float, shape_area float );)r* geom_schemaz:INSERT INTO nybb VALUES ({}, %s, %s, %s, %s );r r r r )formatcursorr'Ziterrowsr!closecommit) r)rr$r*r-Z geom_insertr/sqlir"rrrcreate_postgis\s:       r4)F)Nr,)os.pathosZpandasrZ geopandasrZgeopandas.testingrrrpathabspathdirname__file__ZHEREZ PACKAGE_DIRZunittestr ImportErrorrr r+r4rrrrs    -