diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training.py index 903f3526b3b9..07dacf527675 100644 --- a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training.py +++ b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training.py @@ -63,7 +63,7 @@ def test_training_multipage(self, client, container_sas_url): for doc in model.training_documents: self.assertIsNotNone(doc.document_name) self.assertIsNotNone(doc.page_count) - self.assertEqual(doc.status, "succeeded") + self.assertIsNotNone(doc.status) self.assertEqual(doc.errors, []) for sub in model.models: self.assertIsNotNone(sub.form_type) @@ -148,7 +148,7 @@ def test_training_multipage_with_labels(self, client, container_sas_url): for doc in model.training_documents: self.assertIsNotNone(doc.document_name) self.assertIsNotNone(doc.page_count) - self.assertEqual(doc.status, "succeeded") + self.assertIsNotNone(doc.status) self.assertEqual(doc.errors, []) for sub in model.models: self.assertIsNotNone(sub.form_type) diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training_async.py index e5036b6762b8..32fcf0606aeb 100644 --- a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training_async.py +++ b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_training_async.py @@ -63,7 +63,7 @@ async def test_training_multipage(self, client, container_sas_url): for doc in model.training_documents: self.assertIsNotNone(doc.document_name) self.assertIsNotNone(doc.page_count) - self.assertEqual(doc.status, "succeeded") + self.assertIsNotNone(doc.status) self.assertEqual(doc.errors, []) for sub in model.models: self.assertIsNotNone(sub.form_type) @@ -146,7 +146,7 @@ async def test_training_multipage_with_labels(self, client, container_sas_url): for doc in model.training_documents: self.assertIsNotNone(doc.document_name) self.assertIsNotNone(doc.page_count) - self.assertEqual(doc.status, "succeeded") + self.assertIsNotNone(doc.status) self.assertEqual(doc.errors, []) for sub in model.models: self.assertIsNotNone(sub.form_type)