X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/4ca71978269ba00c687481b90fad621758219b98..5d5f40a82f07e7e2a341435aeb45d8619d252525:/test/bdd/steps/check_functions.py diff --git a/test/bdd/steps/check_functions.py b/test/bdd/steps/check_functions.py index 6d25dd18..f214a886 100644 --- a/test/bdd/steps/check_functions.py +++ b/test/bdd/steps/check_functions.py @@ -1,12 +1,43 @@ +# SPDX-License-Identifier: GPL-2.0-only +# +# This file is part of Nominatim. (https://nominatim.org) +# +# Copyright (C) 2022 by the Nominatim developer community. +# For a full list of authors see the git log. """ Collection of assertion functions used for the steps. """ class Almost: - + """ Compares a float value with a certain jitter. + """ def __init__(self, value, offset=0.00001): self.value = value self.offset = offset def __eq__(self, other): return abs(other - self.value) < self.offset + +class Bbox: + """ Comparator for bounding boxes. + """ + def __init__(self, bbox_string): + self.coord = [float(x) for x in bbox_string.split(',')] + + def __contains__(self, item): + if isinstance(item, str): + item = item.split(',') + item = list(map(float, item)) + + if len(item) == 2: + return self.coord[0] <= item[0] <= self.coord[2] \ + and self.coord[1] <= item[1] <= self.coord[3] + + if len(item) == 4: + return item[0] >= self.coord[0] and item[1] <= self.coord[1] \ + and item[2] >= self.coord[2] and item[3] <= self.coord[3] + + raise ValueError("Not a coordinate or bbox.") + + def __str__(self): + return str(self.coord)