From 6413c03d775b890f020a99a8986df069f651507d Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Wed, 13 May 2026 23:12:06 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20Add=20integration=20tests=20for?=
 =?UTF-8?q?=20rule=5Fbased=5Fextract?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: wjohns989 <56205870+wjohns989@users.noreply.github.com>
---
 tests/test_extraction_rules.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/tests/test_extraction_rules.py b/tests/test_extraction_rules.py
index 3764a64..f2a7344 100644
--- a/tests/test_extraction_rules.py
+++ b/tests/test_extraction_rules.py
@@ -72,3 +72,36 @@ def test_file_path_extraction(self):
         result = rule_based_extract("Edit the file src/main.py to fix the bug")
         entity_names = [e.name for e in result.entities]
         assert any("main.py" in n for n in entity_names)
+
+    def test_rule_based_extract_integration(self):
+        text = "I prefer Python. The meeting is on 2025-10-12. Email me at test@example.com."
+        result = rule_based_extract(text)
+
+        # Assert entities
+        entity_names = [e.name for e in result.entities]
+        assert "Python" in entity_names
+        assert "test@example.com" in entity_names
+
+        # Assert preference relations
+        prefers_rels = [r for r in result.relations if r.predicate == "prefers"]
+        assert len(prefers_rels) > 0
+        assert prefers_rels[0].subject == "user"
+        assert prefers_rels[0].object == "Python"
+
+        # Assert temporal_context
+        assert "2025-10-12" in result.temporal_context
+
+    def test_rule_based_extract_dependency_relations(self):
+        text = "Django depends on Python."
+        result = rule_based_extract(text)
+
+        # Assert entities
+        entity_names = [e.name for e in result.entities]
+        assert "Django" in entity_names
+        assert "Python" in entity_names
+
+        # Assert dependency relations
+        depends_rels = [r for r in result.relations if r.predicate == "depends_on"]
+        assert len(depends_rels) > 0
+        assert depends_rels[0].subject == "Django"
+        assert depends_rels[0].object == "Python"