class_docstr: Unit tests for OpenAI Info Callback
class_docstr_tok: ["Unit","tests","for","OpenAI","Info","Callback"]
code: def test_handler_azure_openai(
      self,
      handler: OpenAICallbackHandler,
      model_name: str,
      expected_cost: float,
  ) -> None:
      response = OpenAIObject(
          {
              "usage": OpenAIObject(
                  {
                      "prompt_tokens": 1000,
                      "completion_tokens": 1000,
                      "total_tokens": 2000,
                  }
              ),
              "model": model_name,
          }
      )
      handler(response)
      assert handler.total_cost == expected_cost
  
file_path: pandas-ai/tests/helpers/test_openai_info.py
imports: import pytest
  from pandasai import SmartDataframe
  from pandasai.helpers import OpenAICallbackHandler, get_openai_callback
  import pandas
  from pandasai.llm.openai import OpenAI
label_desc: Use image node_id 4 for calling the TestOpenAIInfo obj's underlying member method code with example usage: obj.test_handler_azure_openai(handler, model_name, expected_cost) without return types