Skip to content

Commit 342257e

Browse files
committed
make test faster, fix few bugs
1 parent 67a7a1b commit 342257e

File tree

3 files changed

+10
-12
lines changed

3 files changed

+10
-12
lines changed

python_bindings/tests/bindings_test_labels.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
class RandomSelfTestCase(unittest.TestCase):
1010
def testRandomSelf(self):
11-
for idx in range(16):
11+
for idx in range(2):
1212
print("\n**** Index save-load test ****\n")
1313

1414
np.random.seed(idx)

python_bindings/tests/bindings_test_pickle.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -124,13 +124,12 @@ def test_space_main(self, space, dim):
124124
class PickleUnitTests(unittest.TestCase):
125125

126126
def setUp(self):
127+
self.ef_construction = 200
128+
self.M = 32
129+
self.ef = 400
127130

128-
self.ef_construction = 725
129-
self.M = 64
130-
self.ef = 725
131-
132-
self.num_elements = 5000
133-
self.num_test_elements = 200
131+
self.num_elements = 1000
132+
self.num_test_elements = 100
134133

135134
self.num_threads = 4
136135
self.k = 25
@@ -143,10 +142,10 @@ def setUp(self):
143142
# i.e., number of values that are (d1-d2)**2>1e-3
144143

145144
def test_inner_product_space(self):
146-
test_space_main(self, 'ip', 48)
145+
test_space_main(self, 'ip', 16)
147146

148147
def test_l2_space(self):
149-
test_space_main(self, 'l2', 153)
148+
test_space_main(self, 'l2', 53)
150149

151150
def test_cosine_space(self):
152-
test_space_main(self, 'cosine', 512)
151+
test_space_main(self, 'cosine', 32)

python_bindings/tests/bindings_test_spaces.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,5 @@ def testRandomSelf(self):
3535
labels, distances = p.knn_query(np.asarray(data2[-1:]), k=5)
3636

3737

38-
diff=np.mean(np.abs(distances-expected_distances))
39-
print(dim,space, diff)
38+
diff=np.mean(np.abs(distances-expected_distances))
4039
self.assertAlmostEqual(diff, 0, delta=1e-3)

0 commit comments

Comments
 (0)