code: def test_init_with_custom_model(self):
      project_id = "test-project"
      location = "northamerica-northeast1"
      custom_model = "code-bison@001"
  
      vertexai_instance = GoogleVertexAI(
          project_id, location, model=custom_model
      )
  
      assert vertexai_instance.model == custom_model
      assert vertexai_instance.project_id == project_id
      assert vertexai_instance.location == location
  
file_path: pandas-ai/tests/llms/test_google_vertexai.py
imports: import pytest
  from pandasai.llm import GoogleVertexAI
  from pandasai.exceptions import UnsupportedModelError
label_desc: Use image node_id 2 for calling the TestGoogleVertexAI obj's underlying member method code with example usage: obj.test_init_with_custom_model() without return types