Adding support for compressed Ensemble GET endpoint + test adjustment

Author: Jason Costello

hypervector/resources/core/ensemble.py

@@ -1,3 +1,6 @@
+import gzip
+import json
+
 import requests
 
 import hypervector
@@ -40,8 +43,9 @@ def to_response(self):
         }
 
     @classmethod
-    def from_get(cls, dictionary):
+    def from_get(cls, response):
         # Return hypervectors on get
+        dictionary = json.loads(gzip.decompress(response.content))
         ensemble_result = EnsembleResult(
             ensemble_uuid=dictionary['ensemble_uuid'],
             hypervectors=dictionary['hypervectors'],

tests/util.py

@@ -1,3 +1,5 @@
+import gzip
+import json
 import uuid
 from pathlib import Path
 from random import randint
@@ -73,11 +75,12 @@ def mocked_resources(mocked_responses):
     # include hypervectors for GET response
     ensemble_on_get = ensemble.to_response()
     ensemble_on_get['hypervectors'] = [randint(1, 10) for _ in range(ensemble_size)]
+    compressed_ensemble_get = gzip.compress(json.dumps(ensemble_on_get).encode('utf-8'))
 
     mocked_responses.add(
         responses.GET,
         f'{hypervector.API_BASE}/ensemble/{ensemble.ensemble_uuid}',
-        json=ensemble_on_get
+        compressed_ensemble_get
     )
 
     # benchmark