def test_clone(self, con):
some_id = 'some id'
some_dict = {
'title': "some title",
'excludeUttWithoutEntities': True,
'searchDictionaries': [],
'files': 'some files',
'chunk': 'PAGE',
'numWorkers': 8
}
some_response = "some id, {'title': 'some title', 'excludeUttWithoutEntities': True, 'numWorkers': 8, " \
"'chunk': 'PAGE', 'searchDictionaries': [], 'files': 'some files'}"
some_list = ['123']
model = Model()
model.with_documents(some_list)
response = Mock()
response.json.return_value = some_dict
con.return_value = response
res = model.clone(some_id)
self.assertEqual(str(res), some_response)
vocabulary.add_entry("Industrials")
vocabulary.add_entry("Quasi-Governments")
vocabulary.add_entry("Governments")
vocabulary.name("Allocations (%)").create()
# 3- Creator Extractor - Regex must have 1 capturing group
extractor = Extractor()
extractor.set_vocabulary(vocabulary.get_id())
extractor.set_validator("^ +(\\d[\\d\\.\\,]+\\d)")
extractor.set_type(Type.DOUBLE)
# 4- Run
model = Model()
model.set_description("test data process")
model.add_extractor(extractor)
model.with_documents(list_of_documents)
model.create()
# 5- Wait to finish
model.wait_for_completion()
# 6- Export Field results
result = Result(model.get_id())
# print(result.read())
for item in result.read():
field_value = item.get_values()[0]
print(f"{item.get_str()} -> {field_value.get_str()}")
# 7- Export raw results to XLSX
result.result_xlsx_exporter("sample.xlsx")
def test_with_documents_arg_error(self):
list_files = "string"
model = Model()
with self.assertRaises(QtArgumentError):
model.with_documents(list_files)
def test_with_documents(self):
list_files = ["file1", "file2"]
model = Model()
model.with_documents(list_files)
self.assertEqual(list_files, model.temp_dictionary['files'])
