wassname/geopandas
1
0
Fork 0
mirror of https://github.com/wassname/geopandas.git synced 2026-09-10 12:00:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Open database connection within a context manager

Browse Source
This commit is contained in:
Jacob Wasserman
2013-10-08 20:35:27 -04:00
parent 5740137dda
commit 11b46b15bc
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_geodataframe.py
+11 -9
View File
@@ -142,19 +142,21 @@ class TestDataFrame(unittest.TestCase):
if not self.run_db_test:
raise unittest.case.SkipTest()
con = psycopg2.connect(dbname='test_geopandas')
sql = "SELECT * FROM nybb;"
df = GeoDataFrame.read_postgis(sql, con)
with psycopg2.connect(dbname='test_geopandas') as con:
sql = "SELECT * FROM nybb;"
df = GeoDataFrame.read_postgis(sql, con)
self._validate_sql(df)
def test_read_postgis_custom_geom_col(self):
if not self.run_db_test:
raise unittest.case.SkipTest()
con = psycopg2.connect(dbname='test_geopandas')
sql = """SELECT
borocode, boroname, shape_leng, shape_area,
geom AS __geometry__
FROM nybb;"""
df = GeoDataFrame.read_postgis(sql, con, geom_col='__geometry__')
with psycopg2.connect(dbname='test_geopandas') as con:
sql = """SELECT
borocode, boroname, shape_leng, shape_area,
geom AS __geometry__
FROM nybb;"""
df = GeoDataFrame.read_postgis(sql, con, geom_col='__geometry__')
self._validate_sql(df)