From 2af4292e4c31e31469ced66b2f62749e4884f141 Mon Sep 17 00:00:00 2001
From: wlwlwlzhang <wlwlwlzhang@tencent.com>
Date: Wed, 6 Sep 2023 21:17:40 +0800
Subject: [PATCH] add test for pool

---
 test/test_dataset.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/test_dataset.py b/test/test_dataset.py
index 1ab7700..9579687 100644
--- a/test/test_dataset.py
+++ b/test/test_dataset.py
@@ -621,6 +621,9 @@ def insert_data():
                 [thread.start() for thread in threads]
                 [thread.join() for thread in threads]
                 t = db[table_name].count()
+                with db as tx:
+                    if table_name in tx:
+                        tx[table_name].drop()
                 assert t == target_num + 1, t
             except SQLAlchemyError as e:
                 if "timeout expired" in str(e) or "Connection refused" in str(e):