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)