From f050090e8dd3794a22d8a038c828517a021ebac5 Mon Sep 17 00:00:00 2001 From: Patrick Hayes Date: Tue, 22 Oct 2019 10:10:28 -0700 Subject: [PATCH] Add tests --- test/orm/test_query.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) 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) -- 2.47.3