use context management when processing Tiger data

[nominatim.git] / test / bdd / steps / check_functions.py

diff --git a/test/bdd/steps/check_functions.py b/test/bdd/steps/check_functions.py
index 6d25dd18d73f0f75c824cabeb3b1b1a5ff34dda0..f214a88627f4fed870c06d2515cd2b9e4f6b4465 100644 (file)
--- 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:
 """
 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
     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)