Fix test import

Signed-off-by: jamshale <[email protected]>
jamshale · Feb 8, 2024 · 869fd3d · 869fd3d
1 parent c6c3ca6
commit 869fd3d
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/aries_cloudagent/anoncreds/tests/test_revocation_setup.py b/aries_cloudagent/anoncreds/tests/test_revocation_setup.py
@@ -2,12 +2,10 @@
 
 import pytest
 
-from ...askar.profile_anon import (
-    AskarAnoncredsProfile,
-)
-from ...core.in_memory.profile import (
-    InMemoryProfile,
-)
+from aries_cloudagent.tests import mock
+
+from ...askar.profile_anon import AskarAnoncredsProfile
+from ...core.in_memory.profile import InMemoryProfile
 from .. import revocation_setup as test_module
 from ..events import (
     CredDefFinishedEvent,
@@ -17,7 +15,6 @@
 )
 from ..models.anoncreds_revocation import RevRegDef, RevRegDefValue
 from ..revocation import AnonCredsRevocation
-from ..tests import mock
 
 
 @pytest.mark.anoncreds