Fix mock import for python3

kkroening · kkroening · commit e5e293fca471 · 2018-11-25T21:42:43.000-06:00
diff --git a/ffmpeg/tests/test_ffmpeg.py b/ffmpeg/tests/test_ffmpeg.py
@@ -3,13 +3,17 @@
 from builtins import range
 from builtins import str
 import ffmpeg
-import mock
 import os
 import pytest
 import random
 import re
 import subprocess
 
+try:
+    import mock  # python 2
+except ImportError:
+    from unittest import mock  # python 3
+
 
 TEST_DIR = os.path.dirname(__file__)
 SAMPLE_DATA_DIR = os.path.join(TEST_DIR, 'sample_data')
