fixup: make tests work

diff --git a/lib/sqlalchemy/sql/expression.py b/lib/sqlalchemy/sql/expression.py
index 4fa9cda00544344f68a5b79ad15cc9afc24fb78b..f3441c752ce847ef8fd40856a569ad3a2b5435a6 100644 (file)
--- a/lib/sqlalchemy/sql/expression.py
+++ b/lib/sqlalchemy/sql/expression.py
@@ -42,6 +42,7 @@ from ._elements_constructors import outparam as outparam
 from ._elements_constructors import over as over
 from ._elements_constructors import text as text
 from ._elements_constructors import true as true
+from ._elements_constructors import try_cast as try_cast
 from ._elements_constructors import tuple_ as tuple_
 from ._elements_constructors import type_coerce as type_coerce
 from ._elements_constructors import within_group as within_group
@@ -99,6 +100,7 @@ from .elements import SavepointClause as SavepointClause
 from .elements import SQLColumnExpression as SQLColumnExpression
 from .elements import TextClause as TextClause
 from .elements import True_ as True_
+from .elements import TryCast as TryCast
 from .elements import Tuple as Tuple
 from .elements import TypeClause as TypeClause
 from .elements import TypeCoerce as TypeCoerce

diff --git a/test/sql/test_compiler.py b/test/sql/test_compiler.py
index b216ba9ab77093d7814ac63bef82bc6f1d977f95..648056cde57f7614453621bd41b9e4612dd69e24 100644 (file)
--- a/test/sql/test_compiler.py
+++ b/test/sql/test_compiler.py
@@ -6046,12 +6046,12 @@ class StringifySpecialTest(fixtures.TestBase):
         )
 
     def test_try_cast(self):
-        metadata = MetaData()
-        t1 = Table("t1", metadata, Column("id", Integer, primary_key=True))
+        t1 = Table("t1", MetaData(), Column("id", Integer, primary_key=True))
+        expr = select(try_cast(t1.c.id, Integer))
 
-        self.assert_compile(
-            select(try_cast(t1.c.id, Integer)),
-            "SELECT TRY_CAST (t1.id AS INTEGER) AS id FROM t1",
+        eq_ignore_whitespace(
+            str(expr),
+            "SELECT TRY_CAST(t1.id AS INTEGER) AS id FROM t1",
         )