def test_detect_field_value(self):
doc = Document()
doc.pk = 'A'
CsvRegexpsFieldDetectionStrategyMock.text_by_doc_id = {
doc.pk:
"""
Collateral: Enigma Corp
Client ref: "Diane" D.O.O.
"""
}
found_entity = CsvRegexpsFieldDetectionStrategyMock.detect_field_value(
logger, doc, doc_field, {})
self.assertIsNone(found_entity)
CsvRegexpsFieldDetectionStrategyMock.text_by_doc_id = {
doc.pk:
"""
Collateral: Family Name (173437)
Client ref: "Diane" D.O.O.
"""
}
found_entity = CsvRegexpsFieldDetectionStrategyMock.detect_field_value(
logger, doc, doc_field, {})
self.assertIsNotNone(found_entity)
def make_documents(cls, doc_ids: List[int]):
texts = TEST_TEXTS
docs: List[Document] = []
text_index = 0
for id in doc_ids:
doc = Document()
doc.pk = id
doc.project_id = 1
doc.documenttext = DocumentText()
doc.documenttext.full_text = texts[text_index]
text_index += 1
docs.append(doc)
return docs
