Remove unneeded code.

googleapis · Dec 3, 2014 · 3c7df9c · 3c7df9c
1 parent 37a1f9a
commit 3c7df9c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/gcloud/datastore/test_dataset.py b/gcloud/datastore/test_dataset.py
@@ -152,11 +152,9 @@ def test_get_entity_path(self):
         self.assertEqual(result['foo'], 'Foo')
 
     def test_get_entity_odd_nonetype(self):
-        from gcloud.datastore.connection import datastore_pb
         DATASET_ID = 'DATASET'
         KIND = 'Kind'
-        entity_pb = datastore_pb.Entity()
-        connection = _Connection(entity_pb)
+        connection = _Connection()
         dataset = self._makeOne(DATASET_ID, connection)
         with self.assertRaises(ValueError):
             dataset.get_entity([KIND])