install mock on 2.7

Zulko · Feb 17, 2020 · ee6bcc2 · ee6bcc2
1 parent d69d585
commit ee6bcc2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -93,7 +93,8 @@ def run_tests(self):
         'coveralls>=1.1,<2.0',
         'pytest-cov>=2.5.1,<3.0',
         'pytest>=3.0.0,<4.0',
-        'requests>=2.8.1,<3.0'
+        'requests>=2.8.1,<3.0',
+        "mock>=3.0.5; python_version<'2.7'", 
     ]
 
 extra_reqs = {

diff --git a/tests/test_VideoClip.py b/tests/test_VideoClip.py
@@ -3,7 +3,10 @@
 from numpy import sin, pi
 
 import pytest
-from unittest.mock import patch, sentinel
+try:
+    from unittest.mock import patch, sentinel
+except ImportError:
+    from mock import patch, sentinel
 import numpy as np
 
 from moviepy.video.VideoClip import VideoClip, ColorClip