X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/65500927c2955860d392f76fa9b2c318ccd716d2..09f16616388490578ddebdb1d8be5ae545cd1d64:/test/bdd/environment.py?ds=sidebyside diff --git a/test/bdd/environment.py b/test/bdd/environment.py index 6411d011..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 = [] @@ -196,6 +211,15 @@ class OSMDataFactory(object): return scene + def clear_grid(self): + self.grid = {} + + def add_grid_node(self, nodeid, x, y): + self.grid[nodeid] = (x, y) + + def grid_node(self, nodeid): + return self.grid.get(nodeid) + def before_all(context): # logging setup