def test_openai_azure(self):
        """Check that cost tracking works for openai azure models."""

        os.environ["OPENAI_API_VERSION"] = "2023-07-01-preview"
        os.environ["OPENAI_API_TYPE"] = "azure"

        from trulens.providers.openai import AzureOpenAI
        from trulens.providers.openai.endpoint import OpenAIEndpoint

        OpenAIEndpoint.delete_instances()

        provider = AzureOpenAI(
            deployment_name=os.environ["AZURE_OPENAI_DEPLOYMENT"],
            api_key=os.environ["AZURE_OPENAI_API_KEY"],
            azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
            api_version=os.environ["OPENAI_API_VERSION"],
        )

        self._test_llm_provider_endpoint(
            provider, with_cost=False
        )  # no cost tracking for azure openai when using Snowflake's deployment
Check that cost tracking works for openai azure models.
tests/e2e/test_endpoints.py