Also register for geography column

Author: yohanboniface

postgis/geometry.py

@@ -62,3 +62,8 @@ def register(cursor):
     oid = cursor.description[0][1]
     GEOMETRY = _ext.new_type((oid, ), "GEOMETRY", Geometry.from_ewkb)
     _ext.register_type(GEOMETRY)
+
+    cursor.execute("SELECT NULL::geography")
+    oid = cursor.description[0][1]
+    GEOGRAPHY = _ext.new_type((oid, ), "GEOGRAPHY", Geometry.from_ewkb)
+    _ext.register_type(GEOGRAPHY)

tests/test_point.py

@@ -90,3 +90,13 @@ def test_point_with_z_can_be_unpacked_to_dict():
     assert data['y'] == 2
     assert data['z'] == 3
     assert len(data) == 3
+
+
+def test_point_with_geography_column(cursor):
+    cursor.execute('CREATE TABLE geography_point ("geom" geography(PointZ))')
+    params = [Point(1, 2, 3, srid=4326)]
+    cursor.execute('INSERT INTO geography_point (geom) VALUES (%s)', params)
+    cursor.execute('SELECT geom FROM geography_point WHERE geom=%s', params)
+    geom = cursor.fetchone()[0]
+    assert geom.coords == (1, 2, 3)
+    cursor.execute('DROP TABLE geography_point')