Skip to content

Commit 0c18e0a

Browse files
committed
Addressed todo
1 parent 28191fc commit 0c18e0a

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

tests/test_sqlalchemy.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -176,13 +176,9 @@ def test_orm(self, engine: Engine) -> None:
176176
session.add(item3)
177177
session.commit()
178178

179-
stmt = select(Item)
179+
stmt = select(Item).order_by(Item.id)
180180
with Session(engine) as session:
181181
items = [v[0] for v in session.execute(stmt).all()]
182-
# TODO improve
183-
assert items[0].id % 3 == 1
184-
assert items[1].id % 3 == 2
185-
assert items[2].id % 3 == 0
186182
assert items[0].embedding == [1.5, 2, 3]
187183
assert items[1].embedding == [4, 5, 6]
188184
assert items[2].embedding is None

0 commit comments

Comments
 (0)