def test_annoy_wrap_indexer(self): with NumpyIndexer(index_filename='wrap-npidx.gz') as a: a.name = 'wrap-npidx' a.add(vec_idx, vec) a.save() index_abspath = a.index_abspath save_abspath = a.save_abspath with BaseIndexer.load_config(os.path.join(cur_dir, 'annoy-wrap.yml')) as b: idx, dist = b.query(query, top_k=4) 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)) with BaseIndexer.load_config(os.path.join(cur_dir, 'nmslib-wrap.yml')) as c: idx, dist = c.query(query, top_k=4) 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(index_abspath, save_abspath)
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('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('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_annoy_wrap_indexer(metas): with NumpyIndexer(index_filename='wrap-npidx.gz', metas=metas) as indexer: indexer.name = 'wrap-npidx' indexer.add(vec_idx, vec) with BaseIndexer.load_config(os.path.join( cur_dir, 'yaml/annoy-wrap.yml')) as indexer: assert isinstance(indexer, AnnoyIndexer) idx, dist = indexer.query(query, top_k=4) assert idx.shape == dist.shape assert idx.shape == (10, 4)
def test_sptag_wrap_indexer(metas): with NumpyIndexer(index_filename='wrap-npidx.gz', metas=metas) as indexer: indexer.name = 'wrap-npidx' indexer.add(vec_idx, vec) indexer.save() with BaseIndexer.load_config(os.path.join( cur_dir, 'yaml/sptag-wrap.yml')) as indexer: assert isinstance(indexer, SptagIndexer) idx, dist = indexer.query(query, top_k=top_k) assert idx.shape == dist.shape assert idx.shape == (num_queries, top_k)
def test_annoy_wrap_indexer(): with NumpyIndexer(index_filename='wrap-npidx.gz') as indexer: indexer.name = 'wrap-npidx' indexer.add(vec_idx, vec) indexer.save() save_abspath = indexer.save_abspath index_abspath = indexer.index_abspath with BaseIndexer.load_config(os.path.join(cur_dir, 'yaml/annoy-wrap.yml')) as indexer: assert isinstance(indexer, AnnoyIndexer) idx, dist = indexer.query(query, top_k=4) global retr_idx if retr_idx is None: retr_idx = idx else: np.testing.assert_almost_equal(retr_idx, idx) assert idx.shape == dist.shape assert idx.shape == (10, 4) rm_files([save_abspath, index_abspath])
def test_annoy_wrap_indexer(self): with NumpyIndexer(index_filename='wrap-npidx.gz') as indexer: indexer.name = 'wrap-npidx' indexer.add(vec_idx, vec) indexer.save() index_abspath = indexer.index_abspath save_abspath = indexer.save_abspath self.add_tmpfile(index_abspath, save_abspath) with BaseIndexer.load_config( os.path.join(cur_dir, 'yaml/annoy-wrap.yml')) as indexer: self.assertIsInstance(indexer, AnnoyIndexer) idx, dist = indexer.query(query, top_k=4) 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))