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