From 7a51d4e4f51a1906bb6f7303f8a51c87a3ebbbb8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Krier?= <ced@b2ck.com>
Date: Sat, 5 Dec 2020 11:53:48 +0100
Subject: [PATCH] Add generic test for check erase party

issue9895
review324571002
---
 tests/test_purchase.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_purchase.py b/tests/test_purchase.py
index c7bda66e52..ff35a0f147 100644
--- a/tests/test_purchase.py
+++ b/tests/test_purchase.py
@@ -10,11 +10,12 @@ from trytond.tests.test_tryton import doctest_teardown, doctest_checker
 from trytond.transaction import Transaction
 from trytond.pool import Pool
 
-from trytond.modules.company.tests import create_company, set_company
+from trytond.modules.company.tests import (
+    create_company, set_company, PartyCompanyCheckEraseMixin)
 from trytond.modules.account.tests import create_chart
 
 
-class PurchaseTestCase(ModuleTestCase):
+class PurchaseTestCase(PartyCompanyCheckEraseMixin, ModuleTestCase):
     'Test Purchase module'
     module = 'purchase'