Add tests

diff --git a/test/orm/test_query.py b/test/orm/test_query.py
index d2c4f5bc7ba25e08dbcf1fa6328dc1aad0dfe883..1f38dbcd207cddaafe82f044d5a7a51742b52f38 100644 (file)
--- a/test/orm/test_query.py
+++ b/test/orm/test_query.py
@@ -96,32 +96,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)
+        assert query.is_single_entity is True
+        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)
+        assert query.is_single_entity is False
+        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()
         )
+        assert query.is_single_entity is False
+        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()
         )
+        assert query.is_single_entity is False
+        row = query.first()
         assert isinstance(row, collections_abc.Sequence)