Update the test case of pgvector store (#516)

Signed-off-by: SimFG <bang.fu@zilliz.com>
zilliztech · Aug 15, 2023 · d6d484d · d6d484d
1 parent 5aa7e1e
commit d6d484d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/gptcache/__init__.py b/gptcache/__init__.py
@@ -1,5 +1,5 @@
 """gptcache version"""
-__version__ = "0.1.39"
+__version__ = "0.1.39.1"
 
 from gptcache.config import Config
 from gptcache.core import Cache

diff --git a/tests/unit_tests/manager/test_pgvector.py b/tests/unit_tests/manager/test_pgvector.py
@@ -32,11 +32,11 @@ def test_normal(self):
         db.mul_add([VectorData(id=size, data=data[0])])
         ret = db.search(data[0])
         print(ret)
-        self.assertIn(ret[0][0], [0, size])
-        self.assertIn(ret[1][0], [0, size])
+        self.assertIn(ret[0][1], [0, size])
+        self.assertIn(ret[1][1], [0, size])
         db.delete([0, 1, 2, 3, 4, 5, size])
         ret = db.search(data[0])
         print(ret)
-        self.assertNotIn(ret[0][0], [0, size])
+        self.assertNotIn(ret[0][1], [0, size])
         db.rebuild()
         db.close()