diff --git a/tests/test_database.py b/tests/test_database.py index db3e998fa3a..5ed92a4d957 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -120,7 +120,7 @@ async def test_database_drop_recreate(self): await conn.aclose() finally: - await self.con.execute('DROP DATABASE test_db_drop;') + await tb.drop_db(self.con, 'test_db_drop') async def test_database_non_exist_template(self): if not self.has_create_database: @@ -141,4 +141,4 @@ async def test_database_non_exist_template(self): await conn.aclose() finally: - await self.con.execute('DROP DATABASE test_tpl;') + await tb.drop_db(self.con, 'test_tpl')