From 1c63b1c80f1f4454f79b69fcb7c528157d7d804a Mon Sep 17 00:00:00 2001 From: collerek Date: Fri, 12 Mar 2021 18:22:59 +0700 Subject: [PATCH] Add yield1 --- tests/test_default_relation_order.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_default_relation_order.py b/tests/test_default_relation_order.py index b034691..b159a54 100644 --- a/tests/test_default_relation_order.py +++ b/tests/test_default_relation_order.py @@ -72,8 +72,10 @@ def create_test_database(): @pytest.fixture(autouse=True, scope="function") async def cleanup(): - await Book.objects.delete(each=True) - await Author.objects.delete(each=True) + yield + async with database: + await Book.objects.delete(each=True) + await Author.objects.delete(each=True) @pytest.mark.asyncio