move connectivity print stmt out of test fn

amygdala · amygdala · commit 80d2473fdae8 · 2020-06-03T09:50:21.000-07:00
diff --git 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
@@ -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

Original file line number	Diff line number	Diff line change
`@@ -27,4 +27,7 @@ def create_client_with_endpoint(gcp_project_id):`
`27`	`27`	`)`
`28`	`28`	`# [END automl_set_endpoint]`
`29`	`29`
	`30`	`+ # do simple test to check client connectivity`
	`31`	`+ print(client.list_datasets())`
	`32`	`+`
`30`	`33`	`return client`