| """ |
| Unit tests for JSONL Knowledge Graph export and validation. |
| """ |
|
|
| import os |
| import tempfile |
| import unittest |
| from src.graph.guideline_kg import GuidelineKnowledgeGraph |
| from src.graph.jsonl_exporter import JSONLKGExporter |
| from src.validation.jsonl_validator import JSONLKGValidator |
|
|
|
|
| class TestJSONLExportAndValidation(unittest.TestCase): |
| def setUp(self): |
| self.exporter = JSONLKGExporter() |
| self.validator = JSONLKGValidator() |
|
|
| def test_export_and_validate_roundtrip(self): |
| kg = GuidelineKnowledgeGraph("RCOG-GTG-45", "Birth After Previous Caesarean Birth", "RCOG") |
| kg.add_guideline_version_node(2, "2015-10") |
|
|
| prov = { |
| "source_file": "gtg_45.md", |
| "file_sha256": "abc123hash", |
| "page_number": 2, |
| "verbatim_statement": "Planned VBAC is contraindicated in uterine rupture.", |
| "raw_text": "Planned VBAC is contraindicated in uterine rupture. **D**", |
| "revision_tag": "New 2015", |
| "extractor_models": { |
| "ner": "gliner-snomed-biomed:v2.0", |
| "linker": "sapbert-snomed-loinc-rxnorm:v2.0", |
| }, |
| } |
|
|
| rec_id = kg.add_recommendation_node( |
| block_id="REC_0001", |
| edition=2, |
| clinical_question="Contraindications?", |
| statement="Planned VBAC is contraindicated in uterine rupture.", |
| evidence_grade="Grade D", |
| modality="CONTRAINDICATED", |
| section_title="Suitability", |
| valid_from="2015-10", |
| valid_to=None, |
| status="Active", |
| provenance=prov, |
| ) |
|
|
| c1 = kg.add_concept_node( |
| concept_id="441097003", |
| preferred_term="Trial of labour after caesarean delivery", |
| vocabulary="SNOMED-CT", |
| category="PROCEDURE", |
| ) |
| c2 = kg.add_concept_node( |
| concept_id="34430009", |
| preferred_term="Rupture of uterus", |
| vocabulary="SNOMED-CT", |
| category="DISORDER", |
| ) |
|
|
| kg.link_recommendation_to_concept(rec_id, c1, "CONTRAINDICATES_PROCEDURE", "Planned VBAC") |
| kg.link_recommendation_to_concept(rec_id, c2, "CONTRAINDICATED_IN", "uterine rupture", negated=True) |
|
|
| with tempfile.NamedTemporaryFile(suffix=".jsonl", delete=False) as tmp: |
| tmp_path = tmp.name |
|
|
| try: |
| triplets = self.exporter.export_from_graph(kg, tmp_path) |
| self.assertGreater(len(triplets), 0) |
|
|
| |
| report = self.validator.validate_file(tmp_path) |
| self.assertTrue(report.is_valid) |
| self.assertEqual(report.invalid_lines, 0) |
| self.assertEqual(report.provenance_completeness_rate, 100.0) |
| self.assertIn("SNOMED-CT", report.vocabulary_breakdown) |
| self.assertIn("CONTRAINDICATED", report.modality_breakdown) |
|
|
| finally: |
| if os.path.exists(tmp_path): |
| os.remove(tmp_path) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
