From e4c0fb64653f6cc8001b13cec1c56f45b03b5aea Mon Sep 17 00:00:00 2001 From: Greg Friedland Date: Fri, 22 Feb 2019 09:47:32 -0800 Subject: [PATCH] more cleanup --- python_bindings/tests/bindings_test.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/python_bindings/tests/bindings_test.py b/python_bindings/tests/bindings_test.py index 1e3ec87..8406033 100644 --- a/python_bindings/tests/bindings_test.py +++ b/python_bindings/tests/bindings_test.py @@ -148,11 +148,8 @@ def testReloadIndex(self): s = self.bit_vector_str_func(np.ones(512)) original_results = original.knnQuery(s) reloaded_results = reloaded.knnQuery(s) - original_results = list(zip(list(original_results[0]), list(original_results[1]))) - original_results = sorted(original_results, key=lambda x: x[1]) - reloaded_results = list(zip(list(reloaded_results[0]), list(reloaded_results[1]))) - reloaded_results = sorted(reloaded_results, key=lambda x: x[1]) - npt.assert_allclose(original_results, reloaded_results) + npt.assert_allclose(original_results, + reloaded_results) class HNSWTestCase(unittest.TestCase, DenseIndexTestMixin): @@ -174,7 +171,7 @@ def _get_index(self, space='jaccard_sparse'): class BitHammingTestCase(unittest.TestCase, BitVectorIndexTestMixin): def _get_index(self, space='bit_hamming'): - return nmslib.init(method='hnsw', space='bit_hamming', data_type=nmslib.DataType.OBJECT_AS_STRING, + return nmslib.init(method='hnsw', space=space, data_type=nmslib.DataType.OBJECT_AS_STRING, dtype=nmslib.DistType.INT)