mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-08-30 04:40:16 +00:00
KaifAhmad1
58707ff721
- Robust ID extraction in CentralityCalculator, CommunityDetector, and ConnectivityAnalyzer - Support for direct Entity objects and dictionaries as node identifiers - Improved Entity hashability in utils/types.py - Added integration test to verify fix and prevent regression
84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
import unittest
|
|
from semantica.reasoning.reasoner import Reasoner, Rule, RuleType, Fact, InferenceResult
|
|
|
|
class TestReasoner(unittest.TestCase):
|
|
def setUp(self):
|
|
self.reasoner = Reasoner()
|
|
|
|
def test_add_rule_string(self):
|
|
rule_str = "IF Person(?x) AND Parent(?x, ?y) THEN Child(?y, ?x)"
|
|
rule = self.reasoner.add_rule(rule_str)
|
|
self.assertEqual(len(self.reasoner.rules), 1)
|
|
self.assertEqual(rule.conditions, ["Person(?x)", "Parent(?x, ?y)"])
|
|
self.assertEqual(rule.conclusion, "Child(?y, ?x)")
|
|
|
|
def test_add_rule_object(self):
|
|
rule = Rule(
|
|
rule_id="r1",
|
|
name="Test Rule",
|
|
conditions=["A(?x)"],
|
|
conclusion="B(?x)",
|
|
priority=10
|
|
)
|
|
self.reasoner.add_rule(rule)
|
|
self.assertEqual(len(self.reasoner.rules), 1)
|
|
self.assertEqual(self.reasoner.rules[0].priority, 10)
|
|
|
|
def test_add_fact_string(self):
|
|
self.reasoner.add_fact("Person(John)")
|
|
self.assertIn("Person(John)", self.reasoner.facts)
|
|
|
|
def test_add_fact_dict_entity(self):
|
|
fact_dict = {"type": "Person", "name": "John"}
|
|
self.reasoner.add_fact(fact_dict)
|
|
self.assertIn("Person(John)", self.reasoner.facts)
|
|
|
|
def test_add_fact_dict_relationship(self):
|
|
fact_dict = {
|
|
"type": "WorksAt",
|
|
"source_name": "John",
|
|
"target_name": "Google"
|
|
}
|
|
self.reasoner.add_fact(fact_dict)
|
|
self.assertIn("WorksAt(John, Google)", self.reasoner.facts)
|
|
|
|
def test_forward_chaining(self):
|
|
self.reasoner.add_rule("IF Person(?x) AND Parent(?x, ?y) THEN Child(?y, ?x)")
|
|
self.reasoner.add_fact("Person(John)")
|
|
self.reasoner.add_fact("Parent(John, Jane)")
|
|
|
|
results = self.reasoner.forward_chain()
|
|
self.assertEqual(len(results), 1)
|
|
self.assertEqual(results[0].conclusion, "Child(Jane, John)")
|
|
self.assertIn("Child(Jane, John)", self.reasoner.facts)
|
|
|
|
def test_backward_chaining_simple(self):
|
|
self.reasoner.add_rule("IF Person(?x) AND Parent(?x, ?y) THEN Child(?y, ?x)")
|
|
self.reasoner.add_fact("Person(John)")
|
|
self.reasoner.add_fact("Parent(John, Jane)")
|
|
|
|
result = self.reasoner.backward_chain("Child(Jane, John)")
|
|
self.assertIsNotNone(result)
|
|
self.assertEqual(result.conclusion, "Child(Jane, John)")
|
|
self.assertEqual(len(result.premises), 2)
|
|
self.assertIn("Person(John)", result.premises)
|
|
self.assertIn("Parent(John, Jane)", result.premises)
|
|
|
|
def test_infer_facts(self):
|
|
facts = ["Person(John)", "Parent(John, Jane)"]
|
|
rules = ["IF Person(?x) AND Parent(?x, ?y) THEN Child(?y, ?x)"]
|
|
|
|
inferred = self.reasoner.infer_facts(facts, rules)
|
|
self.assertEqual(len(inferred), 1)
|
|
self.assertEqual(inferred[0], "Child(Jane, John)")
|
|
|
|
def test_clear_reset(self):
|
|
self.reasoner.add_fact("Fact(1)")
|
|
self.reasoner.add_rule("IF A THEN B")
|
|
self.reasoner.clear()
|
|
self.assertEqual(len(self.reasoner.facts), 0)
|
|
self.assertEqual(len(self.reasoner.rules), 0)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|