From: Nick Crews <nicholas.b.crews@gmail.com>
Date: Wed, 10 May 2023 18:47:03 +0000 (-0800)
Subject: fixup: make tests work
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=73f14ea47df5aa847f78f8b07269ff6fd914f91b;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

fixup: make tests work
---

diff --git a/lib/sqlalchemy/sql/expression.py b/lib/sqlalchemy/sql/expression.py
index 4fa9cda005..f3441c752c 100644
--- 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 b216ba9ab7..648056cde5 100644
--- 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",
         )