X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/fbff4fa2187f14e1aa55f92ff4f9866a769ef808..c8d45972bbda2cdc60013a074bf6f25beea29394:/test/bdd/steps/check_functions.py

diff --git a/test/bdd/steps/check_functions.py b/test/bdd/steps/check_functions.py
index 1a6f08ef..49676896 100644
--- a/test/bdd/steps/check_functions.py
+++ b/test/bdd/steps/check_functions.py
@@ -2,12 +2,14 @@
 #
 # This file is part of Nominatim. (https://nominatim.org)
 #
-# Copyright (C) 2022 by the Nominatim developer community.
+# Copyright (C) 2023 by the Nominatim developer community.
 # For a full list of authors see the git log.
 """
 Collection of assertion functions used for the steps.
 """
 import json
+import math
+import re
 
 class Almost:
     """ Compares a float value with a certain jitter.
@@ -19,6 +21,52 @@ class Almost:
     def __eq__(self, other):
         return abs(other - self.value) < self.offset
 
+
+OSM_TYPE = {'N' : 'node', 'W' : 'way', 'R' : 'relation',
+            'n' : 'node', 'w' : 'way', 'r' : 'relation',
+            'node' : 'n', 'way' : 'w', 'relation' : 'r'}
+
+
+class OsmType:
+    """ Compares an OSM type, accepting both N/R/W and node/way/relation.
+    """
+
+    def __init__(self, value):
+        self.value = value
+
+
+    def __eq__(self, other):
+        return other == self.value or other == OSM_TYPE[self.value]
+
+
+    def __str__(self):
+        return f"{self.value} or {OSM_TYPE[self.value]}"
+
+
+class Field:
+    """ Generic comparator for fields, which looks at the type of the
+        value compared.
+    """
+    def __init__(self, value, **extra_args):
+        self.value = value
+        self.extra_args = extra_args
+
+    def __eq__(self, other):
+        if isinstance(self.value, float):
+            return math.isclose(self.value, float(other), **self.extra_args)
+
+        if self.value.startswith('^'):
+            return re.fullmatch(self.value, str(other))
+
+        if isinstance(other, dict):
+            return other == eval('{' + self.value + '}')
+
+        return str(self.value) == str(other)
+
+    def __str__(self):
+        return str(self.value)
+
+
 class Bbox:
     """ Comparator for bounding boxes.
     """