Distance filter ast node

1 files changed, 34 insertions, 1 deletions

diff --git a/test/query/ast_test/test_filter_.py b/test/query/ast_test/test_filter_.py
index 4f69bdc..9eb92e2 100644
--- a/test/query/ast_test/test_filter_.py
+++ b/test/query/ast_test/test_filter_.py
@@ -15,7 +15,7 @@ from bsfs.utils import URI
 from bsfs.query.ast.filter_ import _Expression, FilterExpression, PredicateExpression
 from bsfs.query.ast.filter_ import _Branch, Any, All
 from bsfs.query.ast.filter_ import _Agg, And, Or
-from bsfs.query.ast.filter_ import Not, Has
+from bsfs.query.ast.filter_ import Not, Has, Distance
 from bsfs.query.ast.filter_ import _Value, Is, Equals, Substring, StartsWith, EndsWith
 from bsfs.query.ast.filter_ import _Bounded, LessThan, GreaterThan
 from bsfs.query.ast.filter_ import Predicate, OneOf
@@ -284,6 +284,39 @@ class TestValue(unittest.TestCase):
             self.assertEqual(cls(f).value, f)
 
 
+class TestDistance(unittest.TestCase):
+    def test_essentials(self):
+        ref = (1,2,3)
+        # comparison
+        self.assertEqual(Distance(ref, 3), Distance(ref, 3))
+        self.assertEqual(hash(Distance(ref, 3)), hash(Distance(ref, 3)))
+        # comparison respects type
+        self.assertNotEqual(Distance(ref, 3), FilterExpression())
+        self.assertNotEqual(hash(Distance(ref, 3)), hash(FilterExpression()))
+        # comparison respects reference
+        self.assertNotEqual(Distance((1,2,3), 3, False), Distance((1,2), 3, False))
+        self.assertNotEqual(hash(Distance((1,2,3), 3, False)), hash(Distance((1,2), 3, False)))
+        self.assertNotEqual(Distance((1,2,3), 3, False), Distance((1,5,3), 3, False))
+        self.assertNotEqual(hash(Distance((1,2,3), 3, False)), hash(Distance((1,5,3), 3, False)))
+        # comparison respects threshold
+        self.assertNotEqual(Distance((1,2,3), 3, False), Distance((1,2,3), 3.1, False))
+        self.assertNotEqual(hash(Distance((1,2,3), 3, False)), hash(Distance((1,2,3), 3.1, False)))
+        # comparison respects strict flag
+        self.assertNotEqual(Distance((1,2,3), 3, False), Distance((1,2,3), 3, True))
+        self.assertNotEqual(hash(Distance((1,2,3), 3, False)), hash(Distance((1,2,3), 3, True)))
+        # string conversion
+        self.assertEqual(str(Distance(ref, 3, False)), 'Distance((1, 2, 3), 3.0, False)')
+        self.assertEqual(repr(Distance(ref, 3, False)), 'Distance((1, 2, 3), 3.0, False)')
+
+    def test_members(self):
+        self.assertEqual(Distance((1,2,3), 3, False).reference, (1,2,3))
+        self.assertEqual(Distance((3,2,1), 3, False).reference, (3,2,1))
+        self.assertEqual(Distance((1,2,3), 3, False).threshold, 3.0)
+        self.assertEqual(Distance((1,2,3), 53.45, False).threshold, 53.45)
+        self.assertEqual(Distance((1,2,3), 3, False).strict, False)
+        self.assertEqual(Distance((1,2,3), 3, True).strict, True)
+
+
 class TestBounded(unittest.TestCase):
     def test_essentials(self):
         # comparison respects type