Importing mock better

This commit is contained in:
Alan Saul 2013-01-31 11:48:32 +00:00
parent fe1c5431df
commit 57a79e0d75
2 changed files with 2 additions and 2 deletions

View file

@ -27,7 +27,7 @@ try:
import numpy
import pylab
except ImportError:
from mock import MagicMock
from unittest.mock import MagicMock
MOCK_MODULES = ['numpy', 'pylab', 'scipy']
for mod_name in MOCK_MODULES: