From 1cbf780c001748be522ae21f401798d28cb25879 Mon Sep 17 00:00:00 2001
From: Greg Friedland <greg@atomwise.com>
Date: Sun, 17 Feb 2019 10:02:17 -0800
Subject: [PATCH] profile multiple conditions

---
 python_bindings/tests/bindings_test.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/python_bindings/tests/bindings_test.py b/python_bindings/tests/bindings_test.py
index b5b1062..f812160 100644
--- 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()}")