Merge pull request #87 from jgeewax/86-fix-dataset-reference

Fix #86 - Entity objects from a query have a proper Dataset reference

Author: jgeewax

gcloud/datastore/query.py

@@ -244,4 +244,5 @@ def fetch(self, limit=None):
     entity_pbs = self.dataset().connection().run_query(
         query_pb=clone.to_protobuf(), dataset_id=self.dataset().id())
 
-    return [Entity.from_protobuf(entity) for entity in entity_pbs]
+    return [Entity.from_protobuf(entity, dataset=self.dataset())
+            for entity in entity_pbs]