diff options
Diffstat (limited to 'tests/test_alchemy.py')
| -rw-r--r-- | tests/test_alchemy.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_alchemy.py b/tests/test_alchemy.py index 4255417..c94e425 100644 --- a/tests/test_alchemy.py +++ b/tests/test_alchemy.py @@ -66,7 +66,7 @@ class SQLAlchemyPkSequenceTestCase(unittest.TestCase): def setUp(self): super(SQLAlchemyPkSequenceTestCase, self).setUp() StandardFactory.reset_sequence(1) - NonIntegerPkFactory.FACTORY_SESSION.rollback() + NonIntegerPkFactory._meta.sqlalchemy_session.rollback() def test_pk_first(self): std = StandardFactory.build() @@ -104,7 +104,7 @@ class SQLAlchemyNonIntegerPkTestCase(unittest.TestCase): def setUp(self): super(SQLAlchemyNonIntegerPkTestCase, self).setUp() NonIntegerPkFactory.reset_sequence() - NonIntegerPkFactory.FACTORY_SESSION.rollback() + NonIntegerPkFactory._meta.sqlalchemy_session.rollback() def test_first(self): nonint = NonIntegerPkFactory.build() |