fix (Azure#18327)

jongio · Feb 9, 2021 · 351cdac · 351cdac
1 parent 1abc9f1
commit 351cdac
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 33 deletions.
diff --git a/sdk/textanalytics/Azure.AI.TextAnalytics/tests/RecognizeEntitiesTests.cs b/sdk/textanalytics/Azure.AI.TextAnalytics/tests/RecognizeEntitiesTests.cs
@@ -81,11 +81,14 @@ public async Task RecognizeEntitiesWithSubCategoryTest()
             TextAnalyticsClient client = GetClient();
             string document = "I had a wonderful trip to Seattle last week.";
 
-            CategorizedEntityCollection entities = await client.RecognizeEntitiesAsync(document);
+            RecognizeEntitiesResultCollection result = await client.RecognizeEntitiesBatchAsync(new List<string>() { document }, options: new TextAnalyticsRequestOptions() { ModelVersion = "2020-04-01" } );
 
-            Assert.GreaterOrEqual(entities.Count, 3);
+            var documentResult = result.FirstOrDefault();
+            Assert.IsFalse(documentResult.HasError);
 
-            foreach (CategorizedEntity entity in entities)
+            Assert.GreaterOrEqual(documentResult.Entities.Count, 3);
+
+            foreach (CategorizedEntity entity in documentResult.Entities)
             {
                 if (entity.Text == "last week")
                     Assert.AreEqual("DateRange", entity.SubCategory);

diff --git a/...ics/tests/SessionRecords/RecognizeEntitiesTests/RecognizeEntitiesWithSubCategoryTest.json b/...ics/tests/SessionRecords/RecognizeEntitiesTests/RecognizeEntitiesWithSubCategoryTest.json
diff --git a/...ests/SessionRecords/RecognizeEntitiesTests/RecognizeEntitiesWithSubCategoryTestAsync.json b/...ests/SessionRecords/RecognizeEntitiesTests/RecognizeEntitiesWithSubCategoryTestAsync.json