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))