OM-344: fixed update test import

openimis · Oct 18, 2024 · 183ee8d · 183ee8d
1 parent 3cd49c8
commit 183ee8d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/worker_voucher/tests/test_gql_group_of_worker_update.py b/worker_voucher/tests/test_gql_group_of_worker_update.py
@@ -11,7 +11,11 @@
     gql_mutation_update_group_of_worker_single,
     gql_mutation_update_group_of_worker_multiple
 )
-from worker_voucher.tests.util import create_test_eu_for_user, create_test_worker
+from worker_voucher.tests.util import (
+    create_test_eu_for_user,
+    create_test_worker,
+    create_test_group_of_worker
+)
 
 
 class GQLGroupOfWorkerUpdateTestCase(TestCase):
@@ -43,7 +47,7 @@ def setUpClass(cls):
 
         cls.gql_client = Client(gql_schema)
         cls.gql_context = cls.GQLContext(cls.user)
-        cls.group = cls.create_test_group_of_worker(cls.user, cls.eu, cls.name)
+        cls.group = create_test_group_of_worker(cls.user, cls.eu, cls.name)
 
     def test_update_group_of_worker_success(self):
         InsureeConfig.reset_validation_settings()