def test_single_recognize_pii_entities_empty_text_input(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text="",
)
def test_single_recognize_pii_entities_non_text_input(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
with self.assertRaises(TypeError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text={"id": "1", "text": "hello world"}
)
def test_single_recognize_pii_entities_none_credentials(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
with self.assertRaises(ValueError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=None,
input_text="My SSN is 555-55-5555",
)
def test_single_recognize_pii_entities_non_text_input(self, resource_group, location, text_analytics_account, text_analytics_account_key):
with self.assertRaises(TypeError):
response = single_recognize_pii_entities(
endpoint=text_analytics_account,
credential=TextAnalyticsAPIKeyCredential(text_analytics_account_key),
input_text={"id": "1", "text": "hello world"}
)
def test_single_recognize_pii_entities_bad_credentials(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
with self.assertRaises(ClientAuthenticationError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential="xxxxxxxxxxxx",
input_text="My SSN is 555-55-5555",
)
def test_single_recognize_pii_entities_empty_text_input(self, resource_group, location, text_analytics_account, text_analytics_account_key):
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=text_analytics_account,
credential=TextAnalyticsAPIKeyCredential(text_analytics_account_key),
input_text="",
)
def test_single_recognize_pii_entities_bad_type_credentials(self, resource_group, location, text_analytics_account, text_analytics_account_key):
with self.assertRaises(TypeError):
response = single_recognize_pii_entities(
endpoint=text_analytics_account,
credential=[],
input_text="My SSN is 555-55-5555",
)
def test_single_recognize_pii_entities_bad_credentials(self, resource_group, location, text_analytics_account, text_analytics_account_key):
with self.assertRaises(ClientAuthenticationError):
response = single_recognize_pii_entities(
endpoint=text_analytics_account,
credential=TextAnalyticsAPIKeyCredential("xxxxxxxxxxxx"),
input_text="My SSN is 555-55-5555",
)
def test_single_recognize_pii_entities_bad_language_hint(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text="My SSN is 555-55-5555",
language="English"
)
def test_single_recognize_pii_entities_bad_language_hint(self, resource_group, location, text_analytics_account, text_analytics_account_key):
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=text_analytics_account,
credential=TextAnalyticsAPIKeyCredential(text_analytics_account_key),
input_text="My SSN is 555-55-5555",
language="English"
)
def test_single_recognize_pii_entities_bad_model_version(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text="Microsoft was founded by Bill Gates.",
language="en",
model_version="old"
)
def test_single_recognize_pii_entities_too_many_chars(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
text = ""
for _ in range(5121):
text += "x"
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text=text,
)
def test_single_recognize_pii_entities_too_many_chars(self, resource_group, location, text_analytics_account, text_analytics_account_key):
text = ""
for _ in range(5121):
text += "x"
with self.assertRaises(HttpResponseError):
response = single_recognize_pii_entities(
endpoint=text_analytics_account,
credential=TextAnalyticsAPIKeyCredential(text_analytics_account_key),
input_text=text,
)
def test_single_recognize_pii_entities_response_hook(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
def callback(resp):
self.assertIsNotNone(resp.statistics)
self.assertIsNotNone(resp.model_version)
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text="My SSN is 555-55-5555",
show_stats=True,
model_version="latest",
response_hook=callback
)
def test_successful_single_recognize_pii_entities(self, resource_group, location, cognitiveservices_account, cognitiveservices_account_key):
response = single_recognize_pii_entities(
endpoint=cognitiveservices_account,
credential=cognitiveservices_account_key,
input_text="My SSN is 555-55-5555",
language="en"
)
self.assertEqual(response.entities[0].text, "555-55-5555")
for entity in response.entities:
self.assertIsNotNone(entity.type)
self.assertIsNotNone(entity.offset)
self.assertIsNotNone(entity.length)
self.assertIsNotNone(entity.score)
def recognize_pii_entities(self):
# [START single_recognize_pii_entities]
from azure.ai.textanalytics import single_recognize_pii_entities
text = "The employee's ABA number is 111000025 and his SSN is 555-55-5555."
result = single_recognize_pii_entities(endpoint=self.endpoint,
credential=self.key,
input_text=text,
language="en")
for entity in result.entities:
print("Entity: {}".format(entity.text))
print("Type: {}".format(entity.type))
print("Confidence Score: {}\n".format(entity.score))
