summaryrefslogtreecommitdiff
path: root/test/orm/test_immediate_load.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/orm/test_immediate_load.py')
-rw-r--r--test/orm/test_immediate_load.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/test/orm/test_immediate_load.py b/test/orm/test_immediate_load.py
index 2fdf1afd9..7efd3436c 100644
--- a/test/orm/test_immediate_load.py
+++ b/test/orm/test_immediate_load.py
@@ -1,10 +1,10 @@
"""basic tests of lazy loaded attributes"""
-from sqlalchemy.orm import create_session
from sqlalchemy.orm import immediateload
from sqlalchemy.orm import mapper
from sqlalchemy.orm import relationship
from sqlalchemy.testing import eq_
+from sqlalchemy.testing.fixtures import fixture_session
from test.orm import _fixtures
@@ -22,7 +22,7 @@ class ImmediateTest(_fixtures.FixtureTest):
mapper(Address, addresses)
mapper(User, users, properties={"addresses": relationship(Address)})
- sess = create_session()
+ sess = fixture_session()
result = (
sess.query(User)
@@ -58,7 +58,7 @@ class ImmediateTest(_fixtures.FixtureTest):
users,
properties={"addresses": relationship(Address, lazy="immediate")},
)
- sess = create_session()
+ sess = fixture_session()
result = sess.query(User).filter(users.c.id == 7).all()
eq_(len(sess.identity_map), 2)