sqlamodel tests: Move setup() to __init__.py

flask_admin/tests/sqlamodel/__init__.py

+from flask import Flask
+from flask.ext.admin import Admin
+from flask.ext.sqlalchemy import SQLAlchemy
+
+def setup():
+    app = Flask(__name__)
+    app.config['SECRET_KEY'] = '1'
+    app.config['CSRF_ENABLED'] = False
+    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///'
+
+    db = SQLAlchemy(app)
+    admin = Admin(app)
+
+    return app, db, admin

flask_admin/tests/sqlamodel/test_basic.py

 from nose.tools import eq_, ok_, raises
 
-from flask import Flask
-
 from flask.ext import wtf
-
-from flask.ext.sqlalchemy import SQLAlchemy
-from flask.ext.admin import Admin
 from flask.ext.admin.contrib.sqlamodel import ModelView
 
+from . import setup
 
 class CustomModelView(ModelView):
     def __init__(self, model, session,
@@ -20,7 +16,6 @@ class CustomModelView(ModelView):
                                               name, category,
                                               endpoint, url)
 
-
 def create_models(db):
     class Model1(db.Model):
         def __init__(self, test1=None, test2=None, test3=None, test4=None):
@@ -45,18 +40,6 @@ def create_models(db):
     return Model1, Model2
 
 
-def setup():
-    app = Flask(__name__)
-    app.config['SECRET_KEY'] = '1'
-    app.config['CSRF_ENABLED'] = False
-    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///'
-
-    db = SQLAlchemy(app)
-    admin = Admin(app)
-
-    return app, db, admin
-
-
 def test_model():
     app, db, admin = setup()
     Model1, Model2 = create_models(db)