diff --git a/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare.py b/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare.py index adc9bb560d7b..2f80fdb57802 100644 --- a/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare.py +++ b/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare.py @@ -607,7 +607,7 @@ def test_multiple_pages_of_results_with_errors_returned_successfully(self, clien self.assertIsNotNone(doc.statistics) @GlobalTextAnalyticsAccountPreparer() - @TextAnalyticsClientPreparer(client_kwargs={"text_analytics_account_key": os.environ.get("AZURE_TEXT_ANALYTICS_KEY"), "text_analytics_account": os.environ.get("AZURE_TEXT_ANALYTICS_ENDPOINT")}) + @TextAnalyticsClientPreparer() def test_cancellation(self, client): single_doc = "hello world" docs = [{"id": str(idx), "text": val} for (idx, val) in enumerate(list(itertools.repeat(single_doc, 10)))] @@ -641,7 +641,7 @@ def test_explicit_set_string_index_type(self, client): poller.result() @GlobalTextAnalyticsAccountPreparer() - @TextAnalyticsClientPreparer(client_kwargs={"text_analytics_account_key": os.environ.get("AZURE_TEXT_ANALYTICS_KEY"), "text_analytics_account": os.environ.get("AZURE_TEXT_ANALYTICS_ENDPOINT")}) + @TextAnalyticsClientPreparer() def test_bidirectional_relation_type(self, client): result = list(client.begin_analyze_healthcare_entities( documents=["The patient was diagnosed with Parkinsons Disease (PD)"] diff --git a/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare_async.py b/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare_async.py index 4e0af5eb71a4..4e4ffeb6ef0e 100644 --- a/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare_async.py +++ b/sdk/textanalytics/azure-ai-textanalytics/tests/test_analyze_healthcare_async.py @@ -732,14 +732,14 @@ async def test_multiple_pages_of_results_with_errors_returned_successfully(self, self.assertIsNotNone(doc.statistics) @GlobalTextAnalyticsAccountPreparer() - @TextAnalyticsClientPreparer(client_kwargs={"text_analytics_account_key": os.environ.get("AZURE_TEXT_ANALYTICS_KEY"), "text_analytics_account": os.environ.get("AZURE_TEXT_ANALYTICS_ENDPOINT")}) + @TextAnalyticsClientPreparer() async def test_cancellation(self, client): single_doc = "hello world" docs = [{"id": str(idx), "text": val} for (idx, val) in enumerate(list(itertools.repeat(single_doc, 10)))] async with client: poller = await client.begin_analyze_healthcare_entities(docs, polling_interval=self._interval()) - + try: cancellation_poller = await poller.cancel() cancellation_poller.wait() @@ -768,7 +768,7 @@ async def test_explicit_set_string_index_type(self, client): await poller.result() @GlobalTextAnalyticsAccountPreparer() - @TextAnalyticsClientPreparer(client_kwargs={"text_analytics_account_key": os.environ.get("AZURE_TEXT_ANALYTICS_KEY"), "text_analytics_account": os.environ.get("AZURE_TEXT_ANALYTICS_ENDPOINT")}) + @TextAnalyticsClientPreparer() async def test_bidirectional_relation_type(self, client): response = await (await client.begin_analyze_healthcare_entities( documents=["The patient was diagnosed with Parkinsons Disease (PD)"] @@ -789,4 +789,4 @@ async def test_bidirectional_relation_type(self, client): self.assertEqual(related_entity2, entity1) self.assertEqual(relation_type2, "Abbreviation") - +