use mock from local namespace

diff --git a/tests/test_autogenerate.py b/tests/test_autogenerate.py
index 8566c7f6cf6573838caa16e0fc797b04d295a9c7..3325e7e49817e5b901023d4962e818d35eb25d84 100644 (file)
--- a/tests/test_autogenerate.py
+++ b/tests/test_autogenerate.py
@@ -1,7 +1,7 @@
 import re
 import sys
 from unittest import TestCase
-from mock import Mock, patch
+from . import Mock
 
 from sqlalchemy import MetaData, Column, Table, Integer, String, Text, \
     Numeric, CHAR, ForeignKey, DATETIME, INTEGER, \