def test_init_metadata_store_with_location_override( self, get_metadata_store_mock): aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION) metadata_store._MetadataStore(metadata_store_name=_TEST_ID, location=_TEST_ALT_LOCATION) get_metadata_store_mock.assert_called_once_with( name=_TEST_ALT_LOC_NAME)
def test_init_metadata_store_with_invalid_name(self): with pytest.raises(ValueError): aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION) metadata_store._MetadataStore(metadata_store_name=_TEST_INVALID_NAME)
def test_init_metadata_store_with_id_without_project_or_location(self): with pytest.raises(GoogleAuthError): metadata_store._MetadataStore( metadata_store_name=_TEST_ID, credentials=auth_credentials.AnonymousCredentials(), )
def test_init_metadata_store_with_default_id(self, get_metadata_store_mock): aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION) metadata_store._MetadataStore() get_metadata_store_mock.assert_called_once_with( name=_TEST_DEFAULT_NAME, retry=base._DEFAULT_RETRY )
def test_init_metadata_store(self, get_metadata_store_mock): aiplatform.init(project=_TEST_PROJECT) metadata_store._MetadataStore(metadata_store_name=_TEST_NAME) get_metadata_store_mock.assert_called_once_with( name=_TEST_NAME, retry=base._DEFAULT_RETRY )