test: create test_exists.py

ArVar · Jul 1, 2024 · 86086c6 · 86086c6
1 parent e9fe88f
commit 86086c6
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_exists.py b/tests/test_exists.py
@@ -0,0 +1,13 @@
+import unittest
+
+import numpy
+
+from dedupe.variables.exists import ExistsType
+
+
+class TestExists(unittest.TestCase):
+    def test_comparator(self):
+        var = ExistsType("foo")
+        assert numpy.array_equal(var.comparator(None, None), [0, 0])
+        assert numpy.array_equal(var.comparator(1, 1), [1, 0])
+        assert numpy.array_equal(var.comparator(1, 0), [0, 1])