Skip to content

Commit f7eeb3a

Browse files
committed
Improved test code [skip ci]
1 parent 91f5d34 commit f7eeb3a

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/test_sqlalchemy.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,7 @@ def test_binary_quantize(self):
492492

493493
@pytest.mark.asyncio
494494
@pytest.mark.skipif(sqlalchemy_version == 1, reason='Requires SQLAlchemy 2+')
495-
async def test_async(self):
495+
async def test_async_avg(self):
496496
engine = create_async_engine('postgresql+psycopg://localhost/pgvector_python_test')
497497
async_session = async_sessionmaker(engine, expire_on_commit=False)
498498

@@ -622,8 +622,6 @@ async def test_asyncpg_vector_array(self):
622622
async with async_session() as session:
623623
async with session.begin():
624624
session.add(Item(id=1, embeddings=[np.array([1, 2, 3]), np.array([4, 5, 6])]))
625-
626-
# this fails if the driver does not cast arrays
627625
item = await session.get(Item, 1)
628626
assert item.embeddings[0].tolist() == [1, 2, 3]
629627
assert item.embeddings[1].tolist() == [4, 5, 6]

0 commit comments

Comments
 (0)