@@ -33,8 +33,24 @@ def tearDownClass(cls):
3333 cls .connection .close ()
3434
3535
36+ # Fallback for test runners that doesn't support setUpClass/tearDownClass
37+ def setUp (self ) -> None :
38+ if not hasattr (self , 'connection' ):
39+ self .setUpClass .__func__ (self )
40+ self .private_connection = True
41+
42+ return super ().setUp ()
43+
44+ def tearDown (self ) -> None :
45+ if hasattr (self , 'private_connection' ):
46+ self .tearDownClass .__func__ (self )
47+
48+ return super ().tearDown ()
49+
50+
3651class TestDates (TestWithConnection ):
3752 def setUp (self ):
53+ super ().setUp ()
3854 self .connection .query ("DROP TABLE IF EXISTS a" , None )
3955 self .connection .query ("DROP TABLE IF EXISTS b" , None )
4056 self .preql (
@@ -110,6 +126,7 @@ def test_offset(self):
110126
111127class TestDiffTables (TestWithConnection ):
112128 def setUp (self ):
129+ super ().setUp ()
113130 self .connection .query ("DROP TABLE IF EXISTS ratings_test" , None )
114131 self .connection .query ("DROP TABLE IF EXISTS ratings_test2" , None )
115132 self .preql .load ("./tests/setup.pql" )
@@ -221,9 +238,9 @@ def test_diff_sorted_by_key(self):
221238
222239class TestTableSegment (TestWithConnection ):
223240 def setUp (self ) -> None :
241+ super ().setUp ()
224242 self .table = TableSegment (self .connection , ("ratings_test" ,), "id" , "timestamp" )
225243 self .table2 = TableSegment (self .connection , ("ratings_test2" ,), "id" , "timestamp" )
226- return super ().setUp ()
227244
228245 def test_table_segment (self ):
229246 early = datetime .datetime (2021 , 1 , 1 , 0 , 0 )
0 commit comments