mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-08-29 04:26:20 +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
85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
import unittest
|
|
from unittest.mock import MagicMock, patch, mock_open
|
|
from pathlib import Path
|
|
from semantica.seed.seed_manager import SeedDataManager, SeedDataSource, SeedData
|
|
from semantica.utils.exceptions import ProcessingError
|
|
|
|
class TestSeedDataManager(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.manager = SeedDataManager()
|
|
|
|
def test_initialization(self):
|
|
self.assertIsInstance(self.manager, SeedDataManager)
|
|
self.assertEqual(self.manager.sources, {})
|
|
self.assertIsInstance(self.manager.seed_data, SeedData)
|
|
self.assertEqual(self.manager.versions, {})
|
|
|
|
def test_register_source(self):
|
|
name = "test_source"
|
|
format = "csv"
|
|
location = "test.csv"
|
|
|
|
result = self.manager.register_source(name, format, location, entity_type="Person")
|
|
|
|
self.assertTrue(result)
|
|
self.assertIn(name, self.manager.sources)
|
|
source = self.manager.sources[name]
|
|
self.assertIsInstance(source, SeedDataSource)
|
|
self.assertEqual(source.name, name)
|
|
self.assertEqual(source.format, format)
|
|
self.assertEqual(source.location, location)
|
|
self.assertEqual(source.entity_type, "Person")
|
|
self.assertIn(name, self.manager.versions)
|
|
|
|
@patch("pathlib.Path.exists")
|
|
@patch("builtins.open", new_callable=mock_open, read_data="name,age\nAlice,30\nBob,25")
|
|
def test_load_from_csv(self, mock_file, mock_exists):
|
|
mock_exists.return_value = True
|
|
|
|
records = self.manager.load_from_csv("test.csv", entity_type="Person", source_name="test_source")
|
|
|
|
self.assertEqual(len(records), 2)
|
|
self.assertEqual(records[0]["name"], "Alice")
|
|
self.assertEqual(records[0]["age"], "30")
|
|
self.assertEqual(records[0]["entity_type"], "Person")
|
|
self.assertEqual(records[0]["source"], "test_source")
|
|
|
|
mock_file.assert_called_once_with(Path("test.csv"), "r", encoding="utf-8")
|
|
|
|
@patch("pathlib.Path.exists")
|
|
def test_load_from_csv_file_not_found(self, mock_exists):
|
|
mock_exists.return_value = False
|
|
|
|
with self.assertRaises(ProcessingError):
|
|
self.manager.load_from_csv("nonexistent.csv")
|
|
|
|
@patch("semantica.seed.seed_manager.read_json_file")
|
|
@patch("pathlib.Path.exists")
|
|
def test_load_from_json(self, mock_exists, mock_read_json):
|
|
mock_exists.return_value = True
|
|
mock_read_json.return_value = [{"name": "Alice", "age": 30}, {"name": "Bob", "age": 25}]
|
|
|
|
records = self.manager.load_from_json("test.json", entity_type="Person", source_name="test_source")
|
|
|
|
self.assertEqual(len(records), 2)
|
|
self.assertEqual(records[0]["name"], "Alice")
|
|
self.assertEqual(records[0]["age"], 30)
|
|
self.assertEqual(records[0]["entity_type"], "Person")
|
|
self.assertEqual(records[0]["source"], "test_source")
|
|
|
|
@patch("semantica.seed.seed_manager.read_json_file")
|
|
@patch("pathlib.Path.exists")
|
|
def test_load_from_json_dict(self, mock_exists, mock_read_json):
|
|
mock_exists.return_value = True
|
|
mock_read_json.return_value = {"entities": [{"name": "Alice", "age": 30}]}
|
|
|
|
records = self.manager.load_from_json("test.json", entity_type="Person")
|
|
|
|
self.assertEqual(len(records), 1)
|
|
self.assertEqual(records[0]["name"], "Alice")
|
|
self.assertEqual(records[0]["entity_type"], "Person")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|