test: add unit test for quantization

jina-ai · May 1, 2020 · eaa43df · eaa43df
1 parent ed6e54c
commit eaa43df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_quant.py b/tests/test_quant.py
@@ -20,7 +20,8 @@ def random_docs():
         d = jina_pb2.Document()
         for k in range(chunks_per_doc):
             c = d.chunks.add()
-            c.embedding.CopyFrom(array2pb(np.random.random([embed_dim])))
+            # force sending at non-quantization
+            c.embedding.CopyFrom(array2pb(np.random.random([embed_dim]), quantize=None))
             c.chunk_id = c_id
             c.doc_id = j
             c_id += 1