@@ -24,13 +24,13 @@ def pytest_generate_tests(metafunc) -> None:
2424
2525class TestQuery (query_tests .TestQuery ):
2626 @utils .retry_rate_limit
27- def test_query (self , driver , conn , query ) -> None :
28- super ().test_query (driver , conn , query )
27+ def test_query (self , driver , conn , query , query_setup ) -> None :
28+ super ().test_query (driver , conn , query , query_setup )
2929
3030 @utils .retry_rate_limit
31- def test_execute_schema (self , driver , conn , query ) -> None :
32- super ().test_execute_schema (driver , conn , query )
31+ def test_execute_schema (self , driver , conn , query , query_setup ) -> None :
32+ super ().test_execute_schema (driver , conn , query , query_setup )
3333
3434 @utils .retry_rate_limit
35- def test_get_table_schema (self , driver , conn_factory , query ) -> None :
36- super ().test_get_table_schema (driver , conn_factory , query )
35+ def test_get_table_schema (self , driver , conn , query , query_setup ) -> None :
36+ super ().test_get_table_schema (driver , conn , query , query_setup )
0 commit comments