rewrite sys modules mock

vyperlang · Feb 16, 2024 · 2860dfc · 2860dfc
1 parent a76e8b2
commit 2860dfc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/unitary/jupyter/conftest.py b/tests/unitary/jupyter/conftest.py
@@ -9,15 +9,18 @@
 def replace_modules():
     mocked_modules = {}
 
+    anchor = sys.modules
+
     def replace(modules: dict):
+        sys.modules = sys.modules.copy()
         for module, mock in modules.items():
             assert module not in sys.modules
             sys.modules[module] = mock
             mocked_modules[module] = mock
 
     yield replace
-    for m in mocked_modules:
-        sys.modules.pop(m)
+
+    sys.modules = anchor
 
 
 @pytest.fixture()