profile multiple conditions

mldrugdiscovery · Feb 17, 2019 · 1cbf780 · 1cbf780
1 parent c37e33d
commit 1cbf780
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python_bindings/tests/bindings_test.py b/python_bindings/tests/bindings_test.py
@@ -219,9 +219,12 @@ def _get_index(self, space='bit_jaccard'):
         raise NotImplementedError()
 
     def testKnnQuery(self):
-        nbits = 512
+        for num_elems in [30000, 100000, 300000, 1000000]:
+            for nbits in [512, 2048]:
+                self._testKnnQuery(nbits, num_elems)
+
+    def _testKnnQuery(self, nbits, num_elems):
         chunk_size = 10000
-        num_elems = 100000
 
         ps_proc = psutil.Process()
         # print(f"\n{ps_proc.memory_info()}")