mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-26 02:22:13 -04:00
Mike Bayer
580139598f
This is a straight reformat run using black as is, with no edits applied at all. The black run will format code consistently, however in some cases that are prevalent in SQLAlchemy code it produces too-long lines. The too-long lines will be resolved in the following commit that will resolve all remaining flake8 issues including shadowed builtins, long lines, import order, unused imports, duplicate imports, and docstring issues. Change-Id: I7eda77fed3d8e73df84b3651fd6cfcfe858d4dc9
101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
from sqlalchemy.testing import fixtures, is_true, is_false, eq_
|
|
from sqlalchemy import MetaData, Table, Column, Integer, String
|
|
from sqlalchemy import and_, or_, bindparam
|
|
from sqlalchemy.sql.elements import ClauseList, ColumnElement
|
|
from sqlalchemy.sql import operators
|
|
from sqlalchemy.sql import util as sql_util
|
|
|
|
|
|
class CompareClausesTest(fixtures.TestBase):
|
|
def setup(self):
|
|
m = MetaData()
|
|
self.a = Table("a", m, Column("x", Integer), Column("y", Integer))
|
|
|
|
self.b = Table("b", m, Column("y", Integer), Column("z", Integer))
|
|
|
|
def test_compare_clauselist_associative(self):
|
|
|
|
l1 = and_(self.a.c.x == self.b.c.y, self.a.c.y == self.b.c.z)
|
|
|
|
l2 = and_(self.a.c.y == self.b.c.z, self.a.c.x == self.b.c.y)
|
|
|
|
l3 = and_(self.a.c.x == self.b.c.z, self.a.c.y == self.b.c.y)
|
|
|
|
is_true(l1.compare(l1))
|
|
is_true(l1.compare(l2))
|
|
is_false(l1.compare(l3))
|
|
|
|
def test_compare_clauselist_not_associative(self):
|
|
|
|
l1 = ClauseList(
|
|
self.a.c.x, self.a.c.y, self.b.c.y, operator=operators.sub
|
|
)
|
|
|
|
l2 = ClauseList(
|
|
self.b.c.y, self.a.c.x, self.a.c.y, operator=operators.sub
|
|
)
|
|
|
|
is_true(l1.compare(l1))
|
|
is_false(l1.compare(l2))
|
|
|
|
def test_compare_clauselist_assoc_different_operator(self):
|
|
|
|
l1 = and_(self.a.c.x == self.b.c.y, self.a.c.y == self.b.c.z)
|
|
|
|
l2 = or_(self.a.c.y == self.b.c.z, self.a.c.x == self.b.c.y)
|
|
|
|
is_false(l1.compare(l2))
|
|
|
|
def test_compare_clauselist_not_assoc_different_operator(self):
|
|
|
|
l1 = ClauseList(
|
|
self.a.c.x, self.a.c.y, self.b.c.y, operator=operators.sub
|
|
)
|
|
|
|
l2 = ClauseList(
|
|
self.a.c.x, self.a.c.y, self.b.c.y, operator=operators.div
|
|
)
|
|
|
|
is_false(l1.compare(l2))
|
|
|
|
def test_compare_binds(self):
|
|
b1 = bindparam("foo", type_=Integer())
|
|
b2 = bindparam("foo", type_=Integer())
|
|
b3 = bindparam("bar", type_=Integer())
|
|
b4 = bindparam("foo", type_=String())
|
|
|
|
def c1():
|
|
return 5
|
|
|
|
def c2():
|
|
return 6
|
|
|
|
b5 = bindparam("foo", type_=Integer(), callable_=c1)
|
|
b6 = bindparam("foo", type_=Integer(), callable_=c2)
|
|
b7 = bindparam("foo", type_=Integer(), callable_=c1)
|
|
|
|
b8 = bindparam("foo", type_=Integer, value=5)
|
|
b9 = bindparam("foo", type_=Integer, value=6)
|
|
|
|
is_false(b1.compare(b5))
|
|
is_true(b5.compare(b7))
|
|
is_false(b5.compare(b6))
|
|
is_true(b1.compare(b2))
|
|
|
|
# currently not comparing "key", as we often have to compare
|
|
# anonymous names. however we should really check for that
|
|
is_true(b1.compare(b3))
|
|
|
|
is_false(b1.compare(b4))
|
|
is_false(b1.compare(b8))
|
|
is_false(b8.compare(b9))
|
|
is_true(b8.compare(b8))
|
|
|
|
|
|
class MiscTest(fixtures.TestBase):
|
|
def test_column_element_no_visit(self):
|
|
class MyElement(ColumnElement):
|
|
pass
|
|
|
|
eq_(sql_util.find_tables(MyElement(), check_columns=True), [])
|