Fixed failed tests on Kokoro (Spanner + Translate) [(googleapis#1192)](…

…GoogleCloudPlatform/python-docs-samples#1192) * Fixed failed tests on Kokoro (Spanner + Translate) * Update quickstart_test.py
larkee · Nov 3, 2017 · dae800f · dae800f
1 parent 52e04c6
commit dae800f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/samples/samples/quickstart_test.py b/samples/samples/quickstart_test.py
@@ -15,8 +15,6 @@
 import os
 
 from google.cloud import spanner
-import google.cloud.exceptions
-import google.cloud.spanner.client
 import mock
 import pytest
 
@@ -27,13 +25,13 @@
 
 @pytest.fixture
 def patch_instance():
-    original_instance = google.cloud.spanner.client.Client.instance
+    original_instance = spanner.Client.instance
 
     def new_instance(self, unused_instance_name):
         return original_instance(self, SPANNER_INSTANCE)
 
     instance_patch = mock.patch(
-        'google.cloud.spanner.client.Client.instance',
+        'google.cloud.spanner.Client.instance',
         side_effect=new_instance,
         autospec=True)
 

diff --git a/samples/samples/snippets.py b/samples/samples/snippets.py
@@ -194,7 +194,7 @@ def query_data_with_index(
         ALTER TABLE Albums ADD COLUMN MarketingBudget INT64
 
     """
-    from google.cloud.proto.spanner.v1 import type_pb2
+    from google.cloud.spanner_v1.proto import type_pb2
 
     spanner_client = spanner.Client()
     instance = spanner_client.instance(instance_id)