From 80d2473fdae823b6d3404a39c105de1fb21a5775 Mon Sep 17 00:00:00 2001 From: Amy Unruh Date: Wed, 3 Jun 2020 08:41:48 -0700 Subject: [PATCH] move connectivity print stmt out of test fn --- tables/automl/automl_tables_set_endpoint.py | 3 +++ tables/automl/endpoint_test.py | 3 +-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tables/automl/automl_tables_set_endpoint.py b/tables/automl/automl_tables_set_endpoint.py index fd865c9cac4d..d6ab898b4f5d 100644 --- a/tables/automl/automl_tables_set_endpoint.py +++ b/tables/automl/automl_tables_set_endpoint.py @@ -27,4 +27,7 @@ def create_client_with_endpoint(gcp_project_id): ) # [END automl_set_endpoint] + # do simple test to check client connectivity + print(client.list_datasets()) + return client diff --git a/tables/automl/endpoint_test.py b/tables/automl/endpoint_test.py index e00c24d9d692..b556a4c9827c 100644 --- a/tables/automl/endpoint_test.py +++ b/tables/automl/endpoint_test.py @@ -21,7 +21,6 @@ def test_client_creation(capsys): - client = automl_tables_set_endpoint.create_client_with_endpoint(PROJECT) - print(client.list_datasets()) + automl_tables_set_endpoint.create_client_with_endpoint(PROJECT) out, _ = capsys.readouterr() assert "GRPCIterator" in out