@@ -512,12 +512,12 @@ def setUp(self):
512
512
("d" , 12 ),
513
513
("e" , 9 ),
514
514
]
515
- self .query = ( """
515
+ self .query = """
516
516
select x, %s(y) over (
517
517
order by x rows between 1 preceding and 1 following
518
518
) as sum_y
519
519
from test order by x
520
- """ )
520
+ """
521
521
self .con .create_window_function ("sumint" , 1 , WindowSumInt )
522
522
523
523
def test_win_sum_int (self ):
@@ -531,7 +531,7 @@ def test_win_error_on_create(self):
531
531
532
532
@with_tracebacks (BadWindow )
533
533
def test_win_exception_in_method (self ):
534
- for meth in [ "__init__" , "step" , "value" , "inverse" ] :
534
+ for meth in "__init__" , "step" , "value" , "inverse" :
535
535
with self .subTest (meth = meth ):
536
536
with patch .object (WindowSumInt , meth , side_effect = BadWindow ):
537
537
name = f"exc_{ meth } "
@@ -578,7 +578,6 @@ def finalize(self): return 42
578
578
with self .subTest (meth = meth , cls = cls ):
579
579
name = f"exc_{ meth } "
580
580
self .con .create_window_function (name , 1 , cls )
581
- self .addCleanup (self .con .create_window_function , name , 1 , None )
582
581
with self .assertRaisesRegex (sqlite .OperationalError ,
583
582
f"'{ meth } ' method not defined" ):
584
583
self .cur .execute (self .query % name )
@@ -596,7 +595,6 @@ def inverse(self, x): pass
596
595
597
596
name = "missing_finalize"
598
597
self .con .create_window_function (name , 1 , MissingFinalize )
599
- self .addCleanup (self .con .create_window_function , name , 1 , None )
600
598
self .cur .execute (self .query % name )
601
599
self .cur .fetchall ()
602
600
@@ -617,6 +615,7 @@ class ErrorValueReturn:
617
615
def __init__ (self ): pass
618
616
def step (self , x ): pass
619
617
def value (self ): return 1 << 65
618
+
620
619
self .con .create_window_function ("err_val_ret" , 1 , ErrorValueReturn )
621
620
self .assertRaisesRegex (sqlite .DataError , "string or blob too big" ,
622
621
self .cur .execute , self .query % "err_val_ret" )
0 commit comments