@@ -81,7 +81,7 @@ def test_select_specific_columns(trino_connection):
81
81
@pytest .mark .parametrize ('trino_connection' , ['memory' ], indirect = True )
82
82
def test_define_and_create_table (trino_connection ):
83
83
engine , conn = trino_connection
84
- if not engine .dialect .has_schema (engine , "test" ):
84
+ if not engine .dialect .has_schema (conn , "test" ):
85
85
engine .execute (sqla .schema .CreateSchema ("test" ))
86
86
metadata = sqla .MetaData ()
87
87
try :
@@ -109,7 +109,7 @@ def test_define_and_create_table(trino_connection):
109
109
def test_insert (trino_connection ):
110
110
engine , conn = trino_connection
111
111
112
- if not engine .dialect .has_schema (engine , "test" ):
112
+ if not engine .dialect .has_schema (conn , "test" ):
113
113
engine .execute (sqla .schema .CreateSchema ("test" ))
114
114
metadata = sqla .MetaData ()
115
115
try :
@@ -138,7 +138,7 @@ def test_insert(trino_connection):
138
138
@pytest .mark .parametrize ('trino_connection' , ['memory' ], indirect = True )
139
139
def test_insert_multiple_statements (trino_connection ):
140
140
engine , conn = trino_connection
141
- if not engine .dialect .has_schema (engine , "test" ):
141
+ if not engine .dialect .has_schema (conn , "test" ):
142
142
engine .execute (sqla .schema .CreateSchema ("test" ))
143
143
metadata = sqla .MetaData ()
144
144
users = sqla .Table ('users' ,
@@ -322,7 +322,7 @@ def test_cte(trino_connection):
322
322
def test_json_column (trino_connection , json_object ):
323
323
engine , conn = trino_connection
324
324
325
- if not engine .dialect .has_schema (engine , "test" ):
325
+ if not engine .dialect .has_schema (conn , "test" ):
326
326
engine .execute (sqla .schema .CreateSchema ("test" ))
327
327
metadata = sqla .MetaData ()
328
328
@@ -350,7 +350,7 @@ def test_json_column(trino_connection, json_object):
350
350
def test_get_table_comment (trino_connection ):
351
351
engine , conn = trino_connection
352
352
353
- if not engine .dialect .has_schema (engine , "test" ):
353
+ if not engine .dialect .has_schema (conn , "test" ):
354
354
engine .execute (sqla .schema .CreateSchema ("test" ))
355
355
metadata = sqla .MetaData ()
356
356
0 commit comments