15 lines
405 B
Diff
15 lines
405 B
Diff
--- a/test/test_genty.py
|
|
+++ b/test/test_genty.py
|
|
@@ -3,7 +3,10 @@
|
|
from __future__ import unicode_literals
|
|
import functools
|
|
import inspect
|
|
-from mock import patch
|
|
+try:
|
|
+ from unittest.mock import patch
|
|
+except ImportError:
|
|
+ from mock import patch
|
|
import six
|
|
from genty import genty, genty_args, genty_dataset, genty_repeat, genty_dataprovider
|
|
from genty.genty import REPLACE_FOR_PERIOD_CHAR
|