✅ fix tests

2 files changed, 5 insertions, 5 deletions

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index ae27e56d1..da6bcfc30 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -16,7 +16,6 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 # See the file COPYING for details.
-import copy
 import csv
 import json
 from io import StringIO
@@ -25,7 +24,7 @@ import locale
 from django.apps import apps
 from django.conf import settings
 from django.contrib.auth.models import Permission
-from django.core.exceptions import ValidationError, ImproperlyConfigured
+from django.core.exceptions import ValidationError
 from django.core.files.uploadedfile import SimpleUploadedFile
 from django.template.defaultfilters import slugify
 from django.test.client import Client
@@ -46,6 +45,7 @@ from ishtar_common.models import (
 from ishtar_common import forms_common
 from archaeological_operations.tests import OperationInitTest, ImportTest
 from archaeological_operations import models as models_ope
+from archaeological_operations.views import RELATION_FORMSET_EXTRA_FORM
 from archaeological_context_records import models
 
 from ishtar_common.utils import generate_relation_graph
@@ -1425,7 +1425,7 @@ class ContextRecordRelationTest(ContextRecordInit, TestCase):
     def init_data(self):
         nb_crs = 2
         data = {}
-        for idx in range(nb_crs + views.RELATION_FORMSET_EXTRA_FORM):
+        for idx in range(nb_crs + RELATION_FORMSET_EXTRA_FORM):
             data.update({
                 f"form-{idx}-pk": "",
                 f"form-{idx}-right_record": "",
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index e2a59345d..b26ad4bcb 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -2752,7 +2752,7 @@ class CustomFormTest(TestCase, OperationInitTest):
         step = "selec-operation_modification"
         cls_wiz.wizard_post(c, url, step, {"pk": self.operations[0].pk})
 
-        step = "relations-operation_modification"
+        step = "general-operation_modification"
         data = {
             "{}{}-current_step".format(cls_wiz.url_name, cls_wiz.wizard_name): [step],
         }
@@ -2760,7 +2760,7 @@ class CustomFormTest(TestCase, OperationInitTest):
         self.assertNotEqual(response.status_code, 404)
         CustomForm.objects.create(
             name="Test2",
-            form="operation-080-relations",
+            form="operation-010-general",
             available=True,
             apply_to_all=True,
             enabled=False,