def test_annoy_wrap_indexer(self):
a = NumpyIndexer(index_filename='wrap-npidx.gz')
a.name = 'wrap-npidx'
a.add(vec_idx, vec)
a.save()
a.close()
b = BaseIndexer.load_config(os.path.join(cur_dir, 'annoy-wrap.yml'))
idx, dist = b.query(query, top_k=4)
print(idx, dist)
global retr_idx
if retr_idx is None:
retr_idx = idx
else:
np.testing.assert_almost_equal(retr_idx, idx)
self.assertEqual(idx.shape, dist.shape)
self.assertEqual(idx.shape, (10, 4))
c = BaseIndexer.load_config(os.path.join(cur_dir, 'nmslib-wrap.yml'))
idx, dist = c.query(query, top_k=4)
print(idx, dist)
if retr_idx is None:
retr_idx = idx
else:
np.testing.assert_almost_equal(retr_idx, idx)
self.assertEqual(idx.shape, dist.shape)
self.assertEqual(idx.shape, (10, 4))
self.add_tmpfile(a.index_abspath, a.save_abspath)
def test_scipy_indexer(self):
a = NumpyIndexer(index_filename='np.test.gz', backend='scipy')
a.add(vec_idx, vec)
a.save()
a.close()
self.assertTrue(os.path.exists(a.index_abspath))
# a.query(np.array(np.random.random([10, 5]), dtype=np.float32), top_k=4)
b = BaseIndexer.load(a.save_abspath)
idx, dist = b.query(query, top_k=4)
print(idx, dist)
global retr_idx
if retr_idx is None:
retr_idx = idx
else:
np.testing.assert_almost_equal(retr_idx, idx)
self.assertEqual(idx.shape, dist.shape)
self.assertEqual(idx.shape, (10, 4))
self.add_tmpfile(a.index_abspath, a.save_abspath)