Fixing import error of a no longer existing generated pb2.

Caused by merging #1353 before #1329.
googleapis · Jan 15, 2016 · 9455bf2 · 9455bf2
1 parent 896c86f
commit 9455bf2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gcloud/test__helpers.py b/gcloud/test__helpers.py
@@ -546,9 +546,9 @@ def test_set_simple_field(self):
         self.assertTrue(self._callFUT(cluster_pb, 'serve_nodes'))
 
     def test_unset_message_field(self):
+        from google.protobuf.timestamp_pb2 import Timestamp
         from gcloud.bigtable._generated import (
             bigtable_cluster_data_pb2 as data_pb2)
-        from gcloud.bigtable._generated.timestamp_pb2 import Timestamp
         cluster_pb = data_pb2.Cluster()
         # Check that no fields are attached to the protobuf message.
         self.assertEqual(len(cluster_pb._fields), 0)
@@ -561,9 +561,9 @@ def test_unset_message_field(self):
         self.assertFalse(self._callFUT(cluster_pb, 'delete_time'))
 
     def test_set_message_field(self):
+        from google.protobuf.timestamp_pb2 import Timestamp
         from gcloud.bigtable._generated import (
             bigtable_cluster_data_pb2 as data_pb2)
-        from gcloud.bigtable._generated.timestamp_pb2 import Timestamp
 
         timestamp = Timestamp()
         cluster_pb = data_pb2.Cluster(delete_time=timestamp)