From eaa43dfd8e35cfaa2e7c35fec3f0483b9a0fca68 Mon Sep 17 00:00:00 2001
From: Han Xiao <han.xiao@jina.ai>
Date: Fri, 1 May 2020 13:24:44 +0200
Subject: [PATCH] test: add unit test for quantization

---
 tests/test_quant.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_quant.py b/tests/test_quant.py
index 7684e669186bf..aa60bf38a2ec3 100644
--- 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