Fix mock import for python3

This commit is contained in:
Karl Kroening 2018-11-25 21:42:43 -06:00
parent 462e34bab3
commit e5e293fca4

View File

@ -3,13 +3,17 @@ from builtins import bytes
from builtins import range from builtins import range
from builtins import str from builtins import str
import ffmpeg import ffmpeg
import mock
import os import os
import pytest import pytest
import random import random
import re import re
import subprocess import subprocess
try:
import mock # python 2
except ImportError:
from unittest import mock # python 3
TEST_DIR = os.path.dirname(__file__) TEST_DIR = os.path.dirname(__file__)
SAMPLE_DATA_DIR = os.path.join(TEST_DIR, 'sample_data') SAMPLE_DATA_DIR = os.path.join(TEST_DIR, 'sample_data')