test suite deprecation rampage

1 files changed, 3 insertions, 7 deletions

diff --git a/test/ext/assignmapper.py b/test/ext/assignmapper.py
index fd08d8ea6..5a4057018 100644
--- a/test/ext/assignmapper.py
+++ b/test/ext/assignmapper.py
@@ -3,16 +3,12 @@ import warnings
 from sqlalchemy import *
 from sqlalchemy import exceptions
 from sqlalchemy.orm import create_session, clear_mappers, relation, class_mapper
-import sqlalchemy.ext.assignmapper
+from sqlalchemy.ext.assignmapper import assign_mapper
 from sqlalchemy.ext.sessioncontext import SessionContext
 from testlib import *
 
-def assign_mapper(*args, **kw):
-    try:
-        warnings.filterwarnings('ignore', 'assign_mapper is deprecated')
-        sqlalchemy.ext.assignmapper.assign_mapper(*args, **kw)
-    finally:
-        warnings.filterwarnings('always', 'assign_mapper is deprecated')
+
+testing.squelch_deprecation(assign_mapper)
 
 class AssignMapperTest(PersistTest):
     def setUpAll(self):