import patch

BirdVox · Nov 28, 2018 · 0a37170 · 0a37170
1 parent e0ee66f
commit 0a37170
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -6,6 +6,13 @@
 import tempfile
 
 
+try:
+    # python 3.4+ should use builtin unittest.mock not mock package
+    from unittest.mock import patch
+except ImportError:
+    from mock import patch
+
+
 import birdvoxdetect
 from birdvoxdetect.birdvoxdetect_exceptions import BirdVoxDetectError
 from birdvoxdetect.cli import positive_float, get_file_list, run