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
|
import functools
from devtools_testutils import AzureRecordedTestCase, EnvironmentVariableLoader, recorded_by_proxy
from azure.core.credentials import AzureKeyCredential
from azure.ai.textanalytics import TextAnalysisClient
from azure.ai.textanalytics.models import (
MultiLanguageTextInput,
MultiLanguageInput,
TextEntityLinkingInput,
EntityLinkingActionContent,
AnalyzeTextEntityLinkingResult,
EntityLinkingActionResult,
LinkedEntity,
EntityLinkingMatch,
)
TextAnalysisPreparer = functools.partial(
EnvironmentVariableLoader,
"text_analysis",
text_analysis_endpoint="https://Sanitized.cognitiveservices.azure.com/",
text_analysis_key="fake_key",
)
class TestTextAnalysis(AzureRecordedTestCase):
def create_client(self, endpoint: str, key: str) -> TextAnalysisClient:
return TextAnalysisClient(endpoint, AzureKeyCredential(key))
class TestTextAnalysisCase(TestTextAnalysis):
@TextAnalysisPreparer()
@recorded_by_proxy
def test_analyze_text_recognize_linked_entities(self, text_analysis_endpoint, text_analysis_key):
client = self.create_client(text_analysis_endpoint, text_analysis_key)
text_a = (
"Microsoft was founded by Bill Gates with some friends he met at Harvard. One of his friends, Steve "
"Ballmer, eventually became CEO after Bill Gates as well. Steve Ballmer eventually stepped down as "
"CEO of Microsoft, and was succeeded by Satya Nadella. Microsoft originally moved its headquarters "
"to Bellevue, Washington in January 1979, but is now headquartered in Redmond"
)
body = TextEntityLinkingInput(
text_input=MultiLanguageTextInput(
multi_language_inputs=[MultiLanguageInput(id="A", text=text_a, language="en")]
),
action_content=EntityLinkingActionContent(model_version="latest"),
)
# Sync (non-LRO) call
result = client.analyze_text(body=body)
assert result is not None
assert isinstance(result, AnalyzeTextEntityLinkingResult)
assert result.results is not None
assert result.results.documents is not None
for doc in result.results.documents:
assert isinstance(doc, EntityLinkingActionResult)
assert doc.id is not None
assert doc.entities is not None
for linked in doc.entities:
assert isinstance(linked, LinkedEntity)
assert linked.name is not None
assert linked.language is not None
assert linked.data_source is not None
assert linked.url is not None
assert linked.id is not None
assert linked.matches is not None
for match in linked.matches:
assert isinstance(match, EntityLinkingMatch)
assert match.confidence_score is not None
assert match.text is not None
assert match.offset is not None
assert match.length is not None
|
