X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/e04838cc7106a6bb35589231dead41edee0bc408..e4a76f2d7ec202875345528b0bce9f3b83f282ee:/test/bdd/environment.py diff --git a/test/bdd/environment.py b/test/bdd/environment.py index 65afc816..58494deb 100644 --- a/test/bdd/environment.py +++ b/test/bdd/environment.py @@ -40,6 +40,7 @@ class NominatimEnvironment(object): def write_nominatim_config(self, dbname): f = open(self.local_settings_file, 'w') f.write("= 0: - out = self.get_scene_geometry(scene, geom) - elif geom.find(',') < 0: - out = "'POINT(%s)'::geometry" % geom + return "ST_SetSRID(%s, 4326)" % self.get_scene_geometry(scene, geom) + + if geom.find(',') < 0: + out = "POINT(%s)" % self.mk_wkt_point(geom) elif geom.find('(') < 0: - out = "'LINESTRING(%s)'::geometry" % geom + line = ','.join([self.mk_wkt_point(x) for x in geom.split(',')]) + out = "LINESTRING(%s)" % line else: - out = "'POLYGON(%s)'::geometry" % geom + inner = geom.strip('() ') + line = ','.join([self.mk_wkt_point(x) for x in inner.split(',')]) + out = "POLYGON((%s))" % line + + return "ST_SetSRID('%s'::geometry, 4326)" % out - return "ST_SetSRID(%s, 4326)" % out + def mk_wkt_point(self, point): + geom = point.strip() + if geom.find(' ') >= 0: + return geom + else: + pt = self.grid_node(int(geom)) + assert_is_not_none(pt, "Point not found in grid") + return "%f %f" % pt def get_scene_geometry(self, default_scene, name): geoms = []