summaryrefslogtreecommitdiff
path: root/test/ext/declarative/test_mixin.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/ext/declarative/test_mixin.py')
-rw-r--r--test/ext/declarative/test_mixin.py68
1 files changed, 36 insertions, 32 deletions
diff --git a/test/ext/declarative/test_mixin.py b/test/ext/declarative/test_mixin.py
index f51a75b3e..7fec77270 100644
--- a/test/ext/declarative/test_mixin.py
+++ b/test/ext/declarative/test_mixin.py
@@ -1,32 +1,36 @@
-from sqlalchemy.testing import (
- eq_,
- assert_raises,
- assert_raises_message,
- is_,
- expect_warnings,
-)
-from sqlalchemy.ext import declarative as decl
import sqlalchemy as sa
+from sqlalchemy import ForeignKey
+from sqlalchemy import func
+from sqlalchemy import Integer
+from sqlalchemy import select
+from sqlalchemy import String
from sqlalchemy import testing
-from sqlalchemy import Integer, String, ForeignKey, select, func
-from sqlalchemy.testing.schema import Table, Column
-from sqlalchemy.orm import (
- relationship,
- create_session,
- class_mapper,
- configure_mappers,
- clear_mappers,
- deferred,
- column_property,
- Session,
- base as orm_base,
- synonym,
-)
-from sqlalchemy.util import classproperty
-from sqlalchemy.ext.declarative import declared_attr, declarative_base
+from sqlalchemy.ext import declarative as decl
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.ext.declarative import declared_attr
+from sqlalchemy.orm import base as orm_base
+from sqlalchemy.orm import class_mapper
+from sqlalchemy.orm import clear_mappers
+from sqlalchemy.orm import column_property
+from sqlalchemy.orm import configure_mappers
+from sqlalchemy.orm import create_session
+from sqlalchemy.orm import deferred
from sqlalchemy.orm import events as orm_events
-from sqlalchemy.testing import fixtures, mock
+from sqlalchemy.orm import relationship
+from sqlalchemy.orm import Session
+from sqlalchemy.orm import synonym
+from sqlalchemy.testing import assert_raises
+from sqlalchemy.testing import assert_raises_message
+from sqlalchemy.testing import eq_
+from sqlalchemy.testing import expect_warnings
+from sqlalchemy.testing import fixtures
+from sqlalchemy.testing import is_
+from sqlalchemy.testing import mock
+from sqlalchemy.testing.schema import Column
+from sqlalchemy.testing.schema import Table
from sqlalchemy.testing.util import gc_collect
+from sqlalchemy.util import classproperty
+
Base = None
@@ -1319,7 +1323,6 @@ class DeclarativeMixinPropertyTest(
def test_correct_for_proxies(self):
from sqlalchemy.ext.hybrid import hybrid_property
- from sqlalchemy.ext import hybrid
from sqlalchemy import inspect
class Mixin(object):
@@ -1388,7 +1391,8 @@ class DeclarativeMixinPropertyTest(
s = Session()
self.assert_compile(
s.query(Base.data_syn).filter(Base.data_syn == "foo"),
- "SELECT test.data AS test_data FROM test WHERE test.data = :data_1",
+ "SELECT test.data AS test_data FROM test "
+ "WHERE test.data = :data_1",
dialect="default",
)
self.assert_compile(
@@ -1770,12 +1774,12 @@ class DeclaredAttrTest(DeclarativeTestBase, testing.AssertsCompiledSQL):
@declared_attr.cascading
def my_attr(cls):
if decl.has_inherited_table(cls):
- id = Column(ForeignKey("a.my_attr"), primary_key=True)
- asserted["b"].add(id)
+ id_ = Column(ForeignKey("a.my_attr"), primary_key=True)
+ asserted["b"].add(id_)
else:
- id = Column(Integer, primary_key=True)
- asserted["a"].add(id)
- return id
+ id_ = Column(Integer, primary_key=True)
+ asserted["a"].add(id_)
+ return id_
class A(Base, Mixin):
__tablename__ = "a"