File size: 5,456 Bytes
e275025 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 |
"""
Tests for Knowledge Base module
"""
import pytest
from pathlib import Path
import tempfile
from knowledge_base import MedicalTermManager, load_terms_from_file, save_terms_to_file
class TestMedicalTermManager:
"""Tests for MedicalTermManager"""
def test_initialization(self):
"""Test basic initialization"""
manager = MedicalTermManager()
assert manager is not None
assert isinstance(manager.terms, set)
def test_add_term(self):
"""Test adding a term"""
manager = MedicalTermManager()
manager.add_term("МРТ")
assert "МРТ" in manager.terms
def test_normalize_term(self):
"""Test term normalization"""
manager = MedicalTermManager()
# Test removing extra spaces
normalized = manager.normalize_term(" МРТ ")
assert normalized == "МРТ"
normalized = manager.normalize_term("Т1-ВИ режим")
assert normalized == "Т1-ВИ режим"
def test_categorization(self):
"""Test automatic categorization"""
manager = MedicalTermManager()
# Add various terms
manager.add_term("МРТ")
manager.add_term("Т1-ВИ")
manager.add_term("L4-L5")
manager.add_term("грыжа")
manager.add_term("гиперинтенсивный")
# Check categories
assert len(manager.categories['imaging']) > 0
assert len(manager.categories['sequences']) > 0
assert len(manager.categories['anatomy']) > 0
assert len(manager.categories['pathology']) > 0
assert len(manager.categories['modifiers']) > 0
def test_search_term(self):
"""Test term search"""
manager = MedicalTermManager()
manager.add_term("МРТ")
manager.add_term("КТ")
manager.add_term("МСКТ")
# Search for "МРТ"
results = manager.search_term("МРТ")
assert "МРТ" in results
# Search for partial match
results = manager.search_term("КТ")
assert len(results) >= 2 # Should match "КТ" and "МСКТ"
def test_get_category_terms(self):
"""Test getting terms by category"""
manager = MedicalTermManager()
manager.add_term("МРТ")
manager.add_term("КТ")
imaging_terms = manager.get_category_terms('imaging')
assert len(imaging_terms) >= 2
def test_get_all_terms(self):
"""Test getting all terms"""
manager = MedicalTermManager()
manager.add_term("МРТ")
manager.add_term("КТ")
manager.add_term("грыжа")
all_terms = manager.get_all_terms()
assert len(all_terms) == 3
assert "МРТ" in all_terms
def test_get_terms_as_text(self):
"""Test getting terms as formatted text"""
manager = MedicalTermManager()
manager.add_term("МРТ")
manager.add_term("КТ")
text = manager.get_terms_as_text()
assert "МРТ" in text
assert "КТ" in text
assert ", " in text # Default separator
def test_validate_transcription(self):
"""Test transcription validation"""
manager = MedicalTermManager()
manager.add_term("МРТ")
manager.add_term("грыжа")
transcription = "Проведено МРТ. Обнаружена грыжа диска."
validation = manager.validate_transcription(transcription)
assert validation['count'] == 2
assert "МРТ" in validation['found_terms']
assert "грыжа" in validation['found_terms']
def test_statistics(self):
"""Test statistics generation"""
manager = MedicalTermManager()
manager.add_term("МРТ")
manager.add_term("Т1-ВИ")
manager.add_term("грыжа")
stats = manager.get_statistics()
assert stats['total_terms'] == 3
assert 'categories' in stats
class TestTermLoader:
"""Tests for term loading/saving functions"""
def test_load_terms_from_file(self):
"""Test loading terms from a file"""
# Create temporary file
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt', encoding='utf-8') as f:
f.write("МРТ, КТ, МСКТ, грыжа")
temp_file = Path(f.name)
try:
terms = load_terms_from_file(temp_file)
assert len(terms) == 4
assert "МРТ" in terms
assert "КТ" in terms
finally:
temp_file.unlink()
def test_save_terms_to_file(self):
"""Test saving terms to a file"""
terms = ["МРТ", "КТ", "МСКТ"]
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
temp_file = Path(f.name)
try:
result = save_terms_to_file(terms, temp_file)
assert result is True
# Verify file contents
content = temp_file.read_text(encoding='utf-8')
assert "МРТ" in content
assert "КТ" in content
assert "МСКТ" in content
finally:
temp_file.unlink()
if __name__ == "__main__":
pytest.main([__file__, "-v"])
|