call select() directly from sqlalchemy instead of flask-sqlalchemy db…

….select()
PnX-SI · Dec 15, 2023 · 5311147 · 5311147
1 parent 08363b2
commit 5311147
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/utils_flask_sqla/tests/test_qfilter.py b/src/utils_flask_sqla/tests/test_qfilter.py
@@ -1,6 +1,6 @@
 import pytest
 from flask import Flask
-from sqlalchemy import func, and_
+from sqlalchemy import func, and_, select
 
 from flask_sqlalchemy import SQLAlchemy
 
@@ -61,21 +61,19 @@ class TestQfilter:
     def test_qfilter(self, bar):
         assert db.session.scalars(BarModel.where_pk_query(bar.pk)).one_or_none() is bar
         assert (
-            db.session.scalars(db.select(BarModel).where(BarModel.where_pk(bar.pk))).one_or_none()
+            db.session.scalars(select(BarModel).where(BarModel.where_pk(bar.pk))).one_or_none()
             is bar
         )
 
         assert db.session.scalars(BarModel.where_pk_query(bar.pk + 1)).one_or_none() is not bar
         assert (
-            db.session.scalars(
-                db.select(BarModel).where(BarModel.where_pk(bar.pk + 1))
-            ).one_or_none()
+            db.session.scalars(select(BarModel).where(BarModel.where_pk(bar.pk + 1))).one_or_none()
             is not bar
         )
 
         assert (
             db.session.scalars(
-                db.select(BarModel).where(BarModel.where_pk_list(bar.pk))
+                select(BarModel).where(BarModel.where_pk_list(bar.pk))
             ).one_or_none()
             is bar
         )