Improved tests [skip ci]

Author: ankane

tests/test_sqlalchemy.py

@@ -158,14 +158,14 @@ def test_filter_orm(self):
     def test_select(self):
         with Session(engine) as session:
             session.add(Item(embedding=[2, 3, 3]))
-            item = session.query(Item.embedding.l2_distance([1, 1, 1])).first()
-            assert item[0] == 3
+            items = session.query(Item.embedding.l2_distance([1, 1, 1])).first()
+            assert items[0] == 3
 
     def test_select_orm(self):
         with Session(engine) as session:
             session.add(Item(embedding=[2, 3, 3]))
-            item = session.scalars(select(Item.embedding.l2_distance([1, 1, 1]))).all()
-            assert item[0] == 3
+            items = session.scalars(select(Item.embedding.l2_distance([1, 1, 1]))).all()
+            assert items[0] == 3
 
     def test_avg(self):
         with Session(engine) as session:

tests/test_sqlmodel.py

@@ -100,8 +100,8 @@ def test_filter(self):
     def test_select(self):
         with Session(engine) as session:
             session.add(Item(embedding=[2, 3, 3]))
-            item = session.exec(select(Item.embedding.l2_distance([1, 1, 1]))).all()
-            assert item[0] == 3
+            items = session.exec(select(Item.embedding.l2_distance([1, 1, 1]))).all()
+            assert items[0] == 3
 
     def test_avg(self):
         with Session(engine) as session: