From: Patrick Hayes <pfhayes@gmail.com>
Date: Tue, 22 Oct 2019 17:10:28 +0000 (-0700)
Subject: Add tests
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=f050090e8dd3794a22d8a038c828517a021ebac5;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Add tests
---

diff --git a/test/orm/test_query.py b/test/orm/test_query.py
index d2c4f5bc7b..1f38dbcd20 100644
--- 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)