Add unit tests

Author: NathalieCharbel

tests/unit/experimental/components/test_resolver.py

@@ -14,14 +14,16 @@
 #  limitations under the License.
 from unittest.mock import MagicMock, call
 
-import neo4j
 import pytest
 from neo4j_graphrag.experimental.components.resolver import (
+    FuzzyMatchResolver,
     SinglePropertyExactMatchResolver,
     SpaCySemanticMatchResolver,
 )
 from neo4j_graphrag.experimental.components.types import ResolutionStats
 
+import neo4j
+
 
 @pytest.mark.asyncio
 async def test_simple_resolver(driver: MagicMock) -> None:
@@ -160,3 +162,78 @@ async def test_spacy_resolver_match_on_multiple_text_properties(
     assert res.number_of_created_nodes == 1
 
     assert driver.execute_query.call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_fuzzy_match_resolver_no_merge(driver: MagicMock) -> None:
+    driver.execute_query.side_effect = [
+        (
+            [
+                neo4j.Record(
+                    {
+                        "lab": "Person",
+                        "labelCluster": [
+                            {"id": 1, "name": "Alice"},
+                            {"id": 2, "name": "Bob"},
+                        ],
+                    }
+                )
+            ],
+            None,
+            None,
+        )
+    ]
+
+    resolver = FuzzyMatchResolver(driver=driver)
+
+    res = await resolver.run()
+    assert isinstance(res, ResolutionStats)
+    assert res.number_of_nodes_to_resolve == 2
+    assert res.number_of_created_nodes == 0
+
+    assert driver.execute_query.call_count == 1
+
+
+@pytest.mark.asyncio
+async def test_fuzzy_match_resolver_multiple_properties(driver: MagicMock) -> None:
+    driver.execute_query.side_effect = [
+        (
+            [
+                neo4j.Record(
+                    {
+                        "lab": "Person",
+                        "labelCluster": [
+                            {"id": 10, "name": "John Smith", "ssn": "123-45-6789"},
+                            {"id": 11, "name": "Jon Smith", "ssn": "123-45-6789"},
+                        ],
+                    }
+                )
+            ],
+            None,
+            None,
+        ),
+        (
+            [neo4j.Record({"id(node)": 10})],
+            None,
+            None,
+        ),
+    ]
+
+    resolver = FuzzyMatchResolver(driver=driver, resolve_properties=["name", "ssn"])
+
+    res = await resolver.run()
+    assert isinstance(res, ResolutionStats)
+
+    assert res.number_of_nodes_to_resolve == 2
+    assert res.number_of_created_nodes == 1
+
+    assert driver.execute_query.call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_fuzzy_match_resolver_normalization(driver: MagicMock) -> None:
+    # instantiate with a dummy driver
+    resolver = FuzzyMatchResolver(driver=driver)
+
+    sim = resolver.compute_similarity("  ALICE  ", "alice!")
+    assert sim == 1