class AnalyzeDocumentRequestPreparation(PerfStressTest): def __init__(self, arguments): super().__init__(arguments) with open( os.path.abspath( os.path.join(os.path.abspath(__file__), "..", "./../sample_forms/forms/Form_1.jpg")), "rb") as fd: self.document_jpg = fd.read() # read test related env vars formrecognizer_test_endpoint = os.environ[ "FORMRECOGNIZER_TEST_ENDPOINT"] form_recognizer_account_key = os.environ["FORMRECOGNIZER_TEST_API_KEY"] # assign the clients that will be used in the perf tests self.service_client = DocumentAnalysisClient( formrecognizer_test_endpoint, AzureKeyCredential(form_recognizer_account_key)) self.async_service_client = AsyncDocumentAnalysisClient( formrecognizer_test_endpoint, AzureKeyCredential(form_recognizer_account_key)) async def close(self): """This is run after cleanup.""" await self.async_service_client.close() self.service_client.close() await super().close() def run_sync(self): """The synchronous perf test.""" poller = self.service_client.begin_analyze_document( "prebuilt-document", self.document_jpg) assert poller async def run_async(self): """The asynchronous perf test.""" poller = await self.async_service_client.begin_analyze_document( "prebuilt-document", self.document_jpg) assert poller
class AnalyzeDocumentFromUrlRequestPreparation(PerfStressTest): def __init__(self, arguments): super().__init__(arguments) self.document_jpg_url = "https://raw.githubusercontent.com/Azure/azure-sdk-for-python/main/sdk/formrecognizer/azure-ai-formrecognizer/tests/sample_forms/forms/Form_1.jpg" # read test related env vars formrecognizer_test_endpoint = os.environ[ "FORMRECOGNIZER_TEST_ENDPOINT"] form_recognizer_account_key = os.environ["FORMRECOGNIZER_TEST_API_KEY"] # assign the clients that will be used in the perf tests self.service_client = DocumentAnalysisClient( formrecognizer_test_endpoint, AzureKeyCredential(form_recognizer_account_key)) self.async_service_client = AsyncDocumentAnalysisClient( formrecognizer_test_endpoint, AzureKeyCredential(form_recognizer_account_key)) async def close(self): """This is run after cleanup.""" await self.async_service_client.close() self.service_client.close() await super().close() def run_sync(self): """The synchronous perf test.""" poller = self.service_client.begin_analyze_document_from_url( "prebuilt-document", self.document_jpg_url) assert poller async def run_async(self): """The asynchronous perf test.""" poller = await self.async_service_client.begin_analyze_document_from_url( "prebuilt-document", self.document_jpg_url) assert poller