update

zcemycl · Nov 7, 2023 · 637d4fd · 637d4fd
1 parent ac3a5ad
commit 637d4fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_db.py b/tests/test_db.py
@@ -116,9 +116,9 @@ async def test_article_pgvector(test_session_orm):
     sess.add_all(all_a)
     await sess.commit()
 
-    stmt = select(d.article).order_by(
+    stmt = select(d.article.title).order_by(
         d.article.factors.l2_distance(a1.factors)
     )
     a_res = (await sess.execute(stmt)).scalars().all()
     for i in range(len(all_a)):
-        assert a_res[i].title == all_a[i].title
+        assert a_res[i] == all_a[i].title