def test_client_w_override_default_version(): test_client_info = gapic_v1.client_info.ClientInfo() test_client_options = client_options.ClientOptions() client_w_override = utils.ModelClientWithOverride( client_options=test_client_options, client_info=test_client_info, ) assert isinstance( client_w_override._clients[ client_w_override._default_version ].get_model.__self__, model_service_client_default.ModelServiceClient, )
def test_client_w_override_select_version(): test_client_info = gapic_v1.client_info.ClientInfo() test_client_options = client_options.ClientOptions() client_w_override = utils.ModelClientWithOverride( client_options=test_client_options, client_info=test_client_info, ) assert isinstance( client_w_override.select_version(compat.V1BETA1).get_model.__self__, model_service_client_v1beta1.ModelServiceClient, ) assert isinstance( client_w_override.select_version(compat.V1).get_model.__self__, model_service_client_v1.ModelServiceClient, )