update tests (#11534)

Azure · May 20, 2020 · 197c0c5 · 197c0c5
1 parent 9285871
commit 197c0c5
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 8 deletions.
diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt.py
@@ -344,8 +344,7 @@ def test_receipt_multipage(self, resource_group, location, form_recognizer_accou
         self.assertEqual(receipt.merchant_name.value, 'Bilbo Baggins')
         self.assertEqual(receipt.merchant_phone_number.value, '+15555555555')
         self.assertEqual(receipt.subtotal.value, 300.0)
-        # TODO: revert after service side fix
-        self.assertIsNotNone(receipt.total.value)
+        self.assertEqual(receipt.total.value, 100.0)
         self.assertEqual(receipt.page_range.first_page, 1)
         self.assertEqual(receipt.page_range.last_page, 1)
         self.assertFormPagesHasValues(receipt.pages)

diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt_async.py
@@ -338,8 +338,7 @@ async def test_receipt_multipage(self, resource_group, location, form_recognizer
         self.assertEqual(receipt.merchant_name.value, 'Bilbo Baggins')
         self.assertEqual(receipt.merchant_phone_number.value, '+15555555555')
         self.assertEqual(receipt.subtotal.value, 300.0)
-        # TODO: revert after service side fix
-        self.assertIsNotNone(receipt.total.value)
+        self.assertEqual(receipt.total.value, 100.0)
         self.assertEqual(receipt.page_range.first_page, 1)
         self.assertEqual(receipt.page_range.last_page, 1)
         self.assertFormPagesHasValues(receipt.pages)

diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt_from_url.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt_from_url.py
@@ -263,8 +263,7 @@ def test_receipt_multipage_url(self, resource_group, location, form_recognizer_a
         self.assertEqual(receipt.merchant_name.value, 'Bilbo Baggins')
         self.assertEqual(receipt.merchant_phone_number.value, '+15555555555')
         self.assertEqual(receipt.subtotal.value, 300.0)
-        # TODO: revert after service side fix
-        self.assertIsNotNone(receipt.total.value)
+        self.assertEqual(receipt.total.value, 100.0)
         self.assertEqual(receipt.page_range.first_page, 1)
         self.assertEqual(receipt.page_range.last_page, 1)
         self.assertFormPagesHasValues(receipt.pages)

diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt_from_url_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/tests/test_receipt_from_url_async.py
@@ -269,8 +269,7 @@ async def test_receipt_multipage_url(self, resource_group, location, form_recogn
         self.assertEqual(receipt.merchant_name.value, 'Bilbo Baggins')
         self.assertEqual(receipt.merchant_phone_number.value, '+15555555555')
         self.assertEqual(receipt.subtotal.value, 300.0)
-        # TODO: revert after service side fix
-        self.assertIsNotNone(receipt.total.value)
+        self.assertEqual(receipt.total.value, 100.0)
         self.assertEqual(receipt.page_range.first_page, 1)
         self.assertEqual(receipt.page_range.last_page, 1)
         self.assertFormPagesHasValues(receipt.pages)