diff options
Diffstat (limited to 'test/orm/test_query.py')
| -rw-r--r-- | test/orm/test_query.py | 20 |
1 files changed, 14 insertions, 6 deletions
diff --git a/test/orm/test_query.py b/test/orm/test_query.py index d2c4f5bc7..cdbeba138 100644 --- a/test/orm/test_query.py +++ b/test/orm/test_query.py @@ -55,7 +55,9 @@ from sqlalchemy.sql import operators from sqlalchemy.testing import AssertsCompiledSQL from sqlalchemy.testing import fixtures from sqlalchemy.testing import is_ +from sqlalchemy.testing import is_false from sqlalchemy.testing import is_not_ +from sqlalchemy.testing import is_true from sqlalchemy.testing import mock from sqlalchemy.testing.assertions import assert_raises from sqlalchemy.testing.assertions import assert_raises_message @@ -96,32 +98,38 @@ class MiscTest(QueryTest): class OnlyReturnTuplesTest(QueryTest): def test_single_entity_false(self): User = self.classes.User - row = create_session().query(User).only_return_tuples(False).first() + query = create_session().query(User).only_return_tuples(False) + is_true(query.is_single_entity) + row = query.first() assert isinstance(row, User) def test_single_entity_true(self): User = self.classes.User - row = create_session().query(User).only_return_tuples(True).first() + query = create_session().query(User).only_return_tuples(True) + is_false(query.is_single_entity) + row = query.first() assert isinstance(row, collections_abc.Sequence) def test_multiple_entity_false(self): User = self.classes.User - row = ( + query = ( create_session() .query(User.id, User) .only_return_tuples(False) - .first() ) + is_false(query.is_single_entity) + row = query.first() assert isinstance(row, collections_abc.Sequence) def test_multiple_entity_true(self): User = self.classes.User - row = ( + query = ( create_session() .query(User.id, User) .only_return_tuples(True) - .first() ) + is_false(query.is_single_entity) + row = query.first() assert isinstance(row, collections_abc.Sequence) |