finally in test_complex_types

saishreeeee · saishreeeee · commit 17696ddfaf79 · 2025-07-31T11:20:26.000+05:30
Signed-off-by: Sai Shree Pradhan &lt;saishree.pradhan@databricks.com&gt;
diff --git a/tests/e2e/test_complex_types.py b/tests/e2e/test_complex_types.py
@@ -39,9 +39,11 @@ def table_fixture(self, connection_details):
                 )
                 """
             )
-            yield
-            # Clean up the table after the test
-            cursor.execute("DELETE FROM pysql_test_complex_types_table")
+            try:
+                yield
+            finally:
+                # Clean up the table after the test
+                cursor.execute("DELETE FROM pysql_test_complex_types_table")
 
     @pytest.mark.parametrize(
         "field,expected_type",

Original file line number	Diff line number	Diff line change
`@@ -39,9 +39,11 @@ def table_fixture(self, connection_details):`
`39`	`39`	`)`
`40`	`40`	`"""`
`41`	`41`	`)`
`42`		`- yield`
`43`		`- # Clean up the table after the test`
`44`		`- cursor.execute("DELETE FROM pysql_test_complex_types_table")`
	`42`	`+ try:`
	`43`	`+ yield`
	`44`	`+ finally:`
	`45`	`+ # Clean up the table after the test`
	`46`	`+ cursor.execute("DELETE FROM pysql_test_complex_types_table")`
`45`	`47`
`46`	`48`	`@pytest.mark.parametrize(`
`47`	`49`	`"field,expected_type",`