diff options
Diffstat (limited to 'test/ext/test_serializer.py')
| -rw-r--r-- | test/ext/test_serializer.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/ext/test_serializer.py b/test/ext/test_serializer.py index 4080a0044..e252a7f55 100644 --- a/test/ext/test_serializer.py +++ b/test/ext/test_serializer.py @@ -176,7 +176,9 @@ class SerializeTest(AssertsCompiledSQL, fixtures.MappedTest): eq_( q2.join(User.addresses) .filter(Address.email == "ed@bettyboop.com") - .value(func.count(literal_column("*"))), + .enable_eagerloads(False) + .with_entities(func.count(literal_column("*"))) + .scalar(), 1, ) u1 = Session.query(User).get(8) @@ -204,7 +206,7 @@ class SerializeTest(AssertsCompiledSQL, fixtures.MappedTest): ) q2 = serializer.loads(serializer.dumps(q, -1), users.metadata, Session) eq_(q2.all(), [User(name="fred")]) - eq_(list(q2.values(User.id, User.name)), [(9, "fred")]) + eq_(list(q2.with_entities(User.id, User.name)), [(9, "fred")]) @testing.requires.non_broken_pickle def test_query_three(self): @@ -221,8 +223,8 @@ class SerializeTest(AssertsCompiledSQL, fixtures.MappedTest): eq_(q2.all(), [User(name="fred")]) # try to pull out the aliased entity here... - ua_2 = q2._entities[0].entity_zero.entity - eq_(list(q2.values(ua_2.id, ua_2.name)), [(9, "fred")]) + ua_2 = q2._compile_state()._entities[0].entity_zero.entity + eq_(list(q2.with_entities(ua_2.id, ua_2.name)), [(9, "fred")]) def test_annotated_one(self): j = join(users, addresses)._annotate({"foo": "bar"}) |