prefer the new unittest.mock from the standard library

kedder · Jun 9, 2024 · 47af854 · 47af854
1 parent 2250754
commit 47af854
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/ofxstatement/tests/test_plugin.py b/src/ofxstatement/tests/test_plugin.py
@@ -1,6 +1,9 @@
 import unittest
 
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 
 import sys
 

diff --git a/src/ofxstatement/tests/test_tool.py b/src/ofxstatement/tests/test_tool.py
@@ -8,7 +8,10 @@
 import shutil
 from typing import Dict
 
-import mock
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 
 from ofxstatement import tool, statement, configuration, parser, exceptions