add compatibility with sqlalchemy 2.0

dropbox · Jun 28, 2023 · fae253b · fae253b
1 parent 183e9e2
commit fae253b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyhive/tests/test_sqlalchemy_trino.py b/pyhive/tests/test_sqlalchemy_trino.py
@@ -86,7 +86,7 @@ def test_reserved_words(self, engine, connection):
         """Trino uses double quotes, not backticks"""
         # Use keywords for the table/column name
         fake_table = Table('select', MetaData(), Column('current_timestamp', String))
-        query = str(fake_table.select().where(fake_table.c.current_timestamp == 'a'))
+        query = str(fake_table.select().where(fake_table.c.current_timestamp == 'a').compile(engine))
         self.assertIn('"select"', query)
         self.assertIn('"current_timestamp"', query)
         self.assertNotIn('`select`', query)