Fix DB-API test rerun issue

tests/test_dbapi_persistence.py

@@ -1,27 +1,40 @@
 #!/usr/bin/env python3
 
+import shutil
 import unittest
 from chdb import dbapi
 
 test_state_dir = ".state_tmp_auxten_dbapi"
 
+
 class TestDBAPIPersistence(unittest.TestCase):
+
+    def setUp(self) -> None:
+        shutil.rmtree(test_state_dir, ignore_errors=True)
+        return super().setUp()
+
+    def tearDown(self):
+        shutil.rmtree(test_state_dir, ignore_errors=True)
+        return super().tearDown()
+
     def test_persistence(self):
         conn = dbapi.connect(path=test_state_dir)
         cur = conn.cursor()
         cur.execute("CREATE DATABASE e ENGINE = Atomic;")
-        cur.execute("CREATE TABLE e.hi (a String primary key, b Int32) Engine = MergeTree ORDER BY a;")
+        cur.execute(
+            "CREATE TABLE e.hi (a String primary key, b Int32) Engine = MergeTree ORDER BY a;"
+        )
         cur.execute("INSERT INTO e.hi (a, b) VALUES (%s, %s);", ["he", 32])
 
         cur.close()
         conn.close()
 
         conn2 = dbapi.connect(path=test_state_dir)
         cur2 = conn2.cursor()
-        cur2.execute('SELECT * FROM e.hi;')
+        cur2.execute("SELECT * FROM e.hi;")
         row = cur2.fetchone()
-        self.assertEqual(('he', 32), row)
+        self.assertEqual(("he", 32), row)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()